milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

#527 update ground truth check

pull/583/head
yudong.cai 2019-11-28 11:27:57 +08:00
parent e3354fd95e
commit a685ab2602
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
core/src/index/unittest/faiss_benchmark/faiss_benchmark_test.cpp
View File

@ -35,13 +35,13 @@
#include <faiss/gpu/StandardGpuResources.h>
#include <faiss/index_io.h>
#ifdef USE_FAISS_V_0_3_0 // faiss_0.3.0
#ifdef USE_FAISS_V_0_3_0 // faiss_0.3.0
#include <faiss/gpu/GpuCloner.h>
#include <faiss/index_factory.h>
#include <faiss/utils/distances.h>
#else // faiss_0.2.1
#else // faiss_0.2.1
#include <faiss/gpu/GpuAutoTune.h>
#include <faiss/utils.h>
@ -210,7 +210,8 @@ GetResultHitCount(const faiss::Index::idx_t* ground_index, const faiss::Index::i
size_t min_k = std::min(ground_k, k);
int hit = 0;
for (int i = 0; i < nq; i++) {
std::set<faiss::Index::idx_t> ground(ground_index + i * ground_k, ground_index + i * ground_k + min_k);
std::set<faiss::Index::idx_t> ground(ground_index + i * ground_k,
ground_index + i * ground_k + min_k / index_add_loops);
for (int j = 0; j < min_k; j++) {
faiss::Index::idx_t id = index[i * k + j];
if (ground.count(id) > 0) {