diff --git a/vectordb_bench/backend/clients/milvus/config.py b/vectordb_bench/backend/clients/milvus/config.py index 70271a600..9ffbdcece 100644 --- a/vectordb_bench/backend/clients/milvus/config.py +++ b/vectordb_bench/backend/clients/milvus/config.py @@ -417,6 +417,7 @@ def search_param(self) -> dict: class SCANNConfig(MilvusIndexConfig, DBCaseConfig): nlist: int = 1024 with_raw_data: bool = False + nprobe: int = 64 reorder_k: int | None = 100 index: IndexType = IndexType.SCANN_MILVUS @@ -433,7 +434,10 @@ def index_param(self) -> dict: def search_param(self) -> dict: return { "metric_type": self.parse_metric(), - "params": {"reorder_k": self.reorder_k}, + "params": { + "nprobe": self.nprobe, + "reorder_k": self.reorder_k, + }, } diff --git a/vectordb_bench/frontend/config/dbCaseConfigs.py b/vectordb_bench/frontend/config/dbCaseConfigs.py index 83a5bb0e0..6d27a0e3d 100644 --- a/vectordb_bench/frontend/config/dbCaseConfigs.py +++ b/vectordb_bench/frontend/config/dbCaseConfigs.py @@ -1056,6 +1056,7 @@ class CaseConfigInput(BaseModel): IndexType.IVFPQ.value, IndexType.IVFSQ8.value, IndexType.IVF_RABITQ.value, + IndexType.SCANN_MILVUS.value, IndexType.GPU_IVF_FLAT.value, IndexType.GPU_IVF_PQ.value, IndexType.GPU_BRUTE_FORCE.value,