admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-07 01:28:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix the case that knng is not full (#6308)

Browse Source 
Signed-off-by: shengjun.li <shengjun.li@zilliz.com>
This commit is contained in:
shengjun.li 2021-07-05 17:26:07 +08:00 committed by GitHub
parent c43938cd8b
commit 1f42b4ec9d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
internal/core/src/index/knowhere/knowhere/index/vector_offset_index/IndexNSG_NM.cpp
View File

@ -134,6 +134,12 @@ NSG_NM::BuildAll(const DatasetPtr& dataset_ptr, const Config& config) {
preprocess_index->GenGraph(raw_data, k, knng, config); preprocess_index->GenGraph(raw_data, k, knng, config);
#endif #endif
for (size_t i = 0; i < knng.size(); i++) {
while (knng[i].size() > 0 && knng[i].back() == -1) {
knng[i].resize(knng[i].size() - 1);
}
}
impl::BuildParams b_params; impl::BuildParams b_params;
b_params.candidate_pool_size = config[IndexParams::candidate]; b_params.candidate_pool_size = config[IndexParams::candidate];
b_params.out_degree = config[IndexParams::out_degree]; b_params.out_degree = config[IndexParams::out_degree];