diff options
author | Arne Juul <arnej@verizonmedia.com> | 2020-03-23 14:07:55 +0000 |
---|---|---|
committer | Arne Juul <arnej@verizonmedia.com> | 2020-03-23 14:30:35 +0000 |
commit | fc6403cca1829df74abc0707f12023fc4857ba26 (patch) | |
tree | 5a08c5ef8336eed7f071fa0ac4d30520d8597a1f /searchcommon | |
parent | b2bef82b64c8ddd707fa58d394eb325aac1812fb (diff) |
add distance metric to HnswIndexParams
Diffstat (limited to 'searchcommon')
-rw-r--r-- | searchcommon/src/vespa/searchcommon/attribute/hnsw_index_params.h | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/searchcommon/src/vespa/searchcommon/attribute/hnsw_index_params.h b/searchcommon/src/vespa/searchcommon/attribute/hnsw_index_params.h index 9e98a8c5fb7..94d3fda49f3 100644 --- a/searchcommon/src/vespa/searchcommon/attribute/hnsw_index_params.h +++ b/searchcommon/src/vespa/searchcommon/attribute/hnsw_index_params.h @@ -2,6 +2,8 @@ #pragma once +#include "distance_metric.h" + namespace search::attribute { /** @@ -12,20 +14,25 @@ class HnswIndexParams { private: uint32_t _max_links_per_node; uint32_t _neighbors_to_explore_at_insert; + DistanceMetric _distance_metric; public: HnswIndexParams(uint32_t max_links_per_node_in, - uint32_t neighbors_to_explore_at_insert_in) + uint32_t neighbors_to_explore_at_insert_in, + DistanceMetric distance_metric_in) : _max_links_per_node(max_links_per_node_in), - _neighbors_to_explore_at_insert(neighbors_to_explore_at_insert_in) + _neighbors_to_explore_at_insert(neighbors_to_explore_at_insert_in), + _distance_metric(distance_metric_in) {} uint32_t max_links_per_node() const { return _max_links_per_node; } uint32_t neighbors_to_explore_at_insert() const { return _neighbors_to_explore_at_insert; } + DistanceMetric distance_metric() const { return _distance_metric; } bool operator==(const HnswIndexParams& rhs) const { - return _max_links_per_node == rhs._max_links_per_node && - _neighbors_to_explore_at_insert == rhs._neighbors_to_explore_at_insert; + return (_max_links_per_node == rhs._max_links_per_node && + _neighbors_to_explore_at_insert == rhs._neighbors_to_explore_at_insert && + _distance_metric == rhs._distance_metric); } }; |