aboutsummaryrefslogtreecommitdiffstats
path: root/searchlib
diff options
context:
space:
mode:
authorArne Juul <arnej@verizonmedia.com>2020-03-02 12:23:06 +0000
committerArne Juul <arnej@verizonmedia.com>2020-03-02 12:23:06 +0000
commit8fde45497e72586a941d81bfac8d8828c6c1ea55 (patch)
tree181a94d4759e11336465d814b2d6999811237738 /searchlib
parentb2d91e1c52fc05a38e808da0ba71fab5a90adedb (diff)
more consistent variable names
Diffstat (limited to 'searchlib')
-rw-r--r--searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.cpp12
-rw-r--r--searchlib/src/vespa/searchlib/queryeval/nearest_neighbor_blueprint.h4
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();