summaryrefslogtreecommitdiffstats
path: root/eval/src/tests/ann/xp-hnsw-wrap.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'eval/src/tests/ann/xp-hnsw-wrap.cpp')
-rw-r--r--eval/src/tests/ann/xp-hnsw-wrap.cpp5
1 files changed, 3 insertions, 2 deletions
diff --git a/eval/src/tests/ann/xp-hnsw-wrap.cpp b/eval/src/tests/ann/xp-hnsw-wrap.cpp
index 33895b2bd7c..3eb01142dcd 100644
--- a/eval/src/tests/ann/xp-hnsw-wrap.cpp
+++ b/eval/src/tests/ann/xp-hnsw-wrap.cpp
@@ -15,7 +15,7 @@ public:
HnswWrapNns(uint32_t numDims, const DocVectorAccess<float> &dva)
: NNS(numDims, dva),
_l2space(numDims),
- _hnsw(&_l2space, 1000000, 16, 200)
+ _hnsw(&_l2space, 2500000, 16, 200)
{
}
@@ -32,7 +32,8 @@ public:
std::vector<NnsHit> topK(uint32_t k, Vector vector, uint32_t search_k) override {
std::vector<NnsHit> reversed;
- auto priQ = _hnsw.searchKnn(vector.cbegin(), std::max(k, search_k));
+ _hnsw.setEf(search_k);
+ auto priQ = _hnsw.searchKnn(vector.cbegin(), k);
while (! priQ.empty()) {
auto pair = priQ.top();
reversed.emplace_back(pair.second, SqDist(pair.first));