admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-04 09:52:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

return search score

Browse Source 
Former-commit-id: 683f5873992e47f5220dbcb3cf7fa0f6dc66e48b
This commit is contained in:
groot 2019-06-09 17:57:11 +08:00
parent cf94a753d7
commit 8ef34c91f4
1 changed files with 19 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
cpp/src/db/scheduler/SearchTaskQueue.cpp
View File

@ -28,6 +28,23 @@ void ClusterResult(const std::vector<long> &output_ids,
}
}
void MergeResult(SearchContext::Id2ScoreMap &score_src,
SearchContext::Id2ScoreMap &score_target,
uint64_t topk) {
for (auto& pair_src : score_src) {
for (auto iter = score_target.begin(); iter != score_target.end(); ++iter) {
if(pair_src.second > iter->second) {
score_target.insert(iter, pair_src);
}
}
}
//remove unused items
while (score_target.size() > topk) {
score_target.pop_back();
}
}
void TopkResult(SearchContext::ResultSet &result_src,
uint64_t topk,
SearchContext::ResultSet &result_target) {
@ -44,20 +61,7 @@ void TopkResult(SearchContext::ResultSet &result_src,
for (size_t i = 0; i < result_src.size(); i++) {
SearchContext::Id2ScoreMap &score_src = result_src[i];
SearchContext::Id2ScoreMap &score_target = result_target[i];
for (auto& pair_src : score_src) {
for (auto iter = score_target.begin(); iter != score_target.end(); ++iter) {
if(pair_src.second > iter->second) {
score_target.insert(iter, pair_src);
}
}
}
//remove unused items
while (score_target.size() > topk) {
auto it_end = score_target.end();
it_end--;
score_target.erase(it_end);
}
MergeResult(score_src, score_target, topk);
}
}
@ -141,7 +145,7 @@ bool SearchTask::DoSearch() {
//step 5: calculate score between 0 ~ 100
CalcScore(context->nq(), context->vectors(), index_engine_->Dimension(), context->GetResult(), result_set);
context->GetResult().swap(result_set);
rc.Record("reduce topk");
rc.Record("calculate score");
//step 6: notify to send result to client
context->IndexSearchDone(index_id_);