blob: 3bc73671463610fe432fe9b3e1ad57d0b42787ef (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "default_nearest_neighbor_index_factory.h"
#include "hnsw_index.h"
#include "random_level_generator.h"
#include "inv_log_level_generator.h"
#include "distance_function_factory.h"
#include <vespa/searchcommon/attribute/config.h>
namespace search::tensor {
using vespalib::eval::ValueType;
namespace {
class LevelZeroGenerator : public RandomLevelGenerator {
uint32_t max_level() override { return 0; }
};
RandomLevelGenerator::UP
make_random_level_generator(uint32_t m)
{
return std::make_unique<InvLogLevelGenerator>(m);
}
} // namespace <unnamed>
std::unique_ptr<NearestNeighborIndex>
DefaultNearestNeighborIndexFactory::make(const DocVectorAccess& vectors,
size_t vector_size,
vespalib::eval::CellType cell_type,
const search::attribute::HnswIndexParams& params) const
{
(void) vector_size;
uint32_t m = params.max_links_per_node();
HnswIndex::Config cfg(m * 2,
m,
params.neighbors_to_explore_at_insert(),
10000,
true);
return std::make_unique<HnswIndex>(vectors,
make_distance_function(params.distance_metric(), cell_type),
make_random_level_generator(m),
cfg);
}
}
|