diff options
author | Arne Juul <arnej@verizonmedia.com> | 2020-03-02 12:23:06 +0000 |
---|---|---|
committer | Arne Juul <arnej@verizonmedia.com> | 2020-03-02 12:23:06 +0000 |
commit | 8fde45497e72586a941d81bfac8d8828c6c1ea55 (patch) | |
tree | 181a94d4759e11336465d814b2d6999811237738 /searchlib | |
parent | b2d91e1c52fc05a38e808da0ba71fab5a90adedb (diff) |
more consistent variable names
Diffstat (limited to 'searchlib')
-rw-r--r-- | searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.cpp | 12 | ||||
-rw-r--r-- | searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.h | 4 |
2 files changed, 9 insertions, 7 deletions
diff --git a/searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.cpp b/searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.cpp index c160f8d5485..06312cbed52 100644 --- a/searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.cpp +++ b/searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.cpp @@ -13,21 +13,21 @@ namespace search::queryeval { NearestNeighborBlueprint::NearestNeighborBlueprint(const queryeval::FieldSpec& field, const tensor::DenseTensorAttribute& attr_tensor, std::unique_ptr<vespalib::tensor::DenseTensorView> query_tensor, - uint32_t target_num_hits, bool approximate, uint32_t explore_k) + uint32_t target_num_hits, bool approximate, uint32_t explore_additional_hits) : ComplexLeafBlueprint(field), _attr_tensor(attr_tensor), _query_tensor(std::move(query_tensor)), _target_num_hits(target_num_hits), _approximate(approximate), - _explore_k(explore_k), + _explore_additional_hits(explore_additional_hits), _distance_heap(target_num_hits), _found_hits() { uint32_t est_hits = _attr_tensor.getNumDocs(); if (_attr_tensor.nearest_neighbor_index()) { est_hits = std::min(target_num_hits, est_hits); - if (_explore_k == 0) { - _explore_k = 100; + if (_explore_additional_hits == 0) { + _explore_additional_hits = 100; } } setEstimate(HitEstimate(est_hits, false)); @@ -46,7 +46,7 @@ NearestNeighborBlueprint::perform_top_k() if (lhs_type == rhs_type) { auto lhs = _query_tensor->cellsRef(); uint32_t k = _target_num_hits; - _found_hits = nns_index->find_top_k(k, lhs, k + _explore_k); + _found_hits = nns_index->find_top_k(k, lhs, k + _explore_additional_hits); } } } @@ -77,6 +77,8 @@ NearestNeighborBlueprint::visitMembers(vespalib::ObjectVisitor& visitor) const visitor.visitString("attribute_tensor", _attr_tensor.getTensorType().to_spec()); visitor.visitString("query_tensor", _query_tensor->type().to_spec()); visitor.visitInt("target_num_hits", _target_num_hits); + visitor.visitBool("approximate", _approximate); + visitor.visitInt("explore_additional_hits", _explore_additional_hits); } bool diff --git a/searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.h b/searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.h index a782633ccc3..39165b066be 100644 --- a/searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.h +++ b/searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.h @@ -22,7 +22,7 @@ private: std::unique_ptr<vespalib::tensor::DenseTensorView> _query_tensor; uint32_t _target_num_hits; bool _approximate; - uint32_t _explore_k; + uint32_t _explore_additional_hits; mutable NearestNeighborDistanceHeap _distance_heap; std::vector<search::tensor::NearestNeighborIndex::Neighbor> _found_hits; @@ -31,7 +31,7 @@ public: NearestNeighborBlueprint(const queryeval::FieldSpec& field, const tensor::DenseTensorAttribute& attr_tensor, std::unique_ptr<vespalib::tensor::DenseTensorView> query_tensor, - uint32_t target_num_hits, bool approximate, uint32_t explore_k); + uint32_t target_num_hits, bool approximate, uint32_t explore_additional_hits); NearestNeighborBlueprint(const NearestNeighborBlueprint&) = delete; NearestNeighborBlueprint& operator=(const NearestNeighborBlueprint&) = delete; ~NearestNeighborBlueprint(); |