Skip to content

Commit 578698a

Browse files
JackLCLchasingegg
authored andcommitted
feat(milvus): add nprobe parameter for SCANN index
- Add nprobe search parameter for SCANN index (default: 64) - Enable nprobe UI configuration for SCANN_MILVUS index type Signed-off-by: JackLCL <chenglong.li@zilliz.com>
1 parent eb2c2ce commit 578698a

2 files changed

Lines changed: 6 additions & 1 deletion

File tree

vectordb_bench/backend/clients/milvus/config.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -417,6 +417,7 @@ def search_param(self) -> dict:
417417
class SCANNConfig(MilvusIndexConfig, DBCaseConfig):
418418
nlist: int = 1024
419419
with_raw_data: bool = False
420+
nprobe: int = 64
420421
reorder_k: int | None = 100
421422
index: IndexType = IndexType.SCANN_MILVUS
422423

@@ -433,7 +434,10 @@ def index_param(self) -> dict:
433434
def search_param(self) -> dict:
434435
return {
435436
"metric_type": self.parse_metric(),
436-
"params": {"reorder_k": self.reorder_k},
437+
"params": {
438+
"nprobe": self.nprobe,
439+
"reorder_k": self.reorder_k,
440+
},
437441
}
438442

439443

vectordb_bench/frontend/config/dbCaseConfigs.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1056,6 +1056,7 @@ class CaseConfigInput(BaseModel):
10561056
IndexType.IVFPQ.value,
10571057
IndexType.IVFSQ8.value,
10581058
IndexType.IVF_RABITQ.value,
1059+
IndexType.SCANN_MILVUS.value,
10591060
IndexType.GPU_IVF_FLAT.value,
10601061
IndexType.GPU_IVF_PQ.value,
10611062
IndexType.GPU_BRUTE_FORCE.value,

0 commit comments

Comments
 (0)