84 lines
1.9 KiB
C++
84 lines
1.9 KiB
C++
/**
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
// -*- c++ -*-
|
|
|
|
#ifndef META_INDEXES_H
|
|
#define META_INDEXES_H
|
|
|
|
#include <faiss/Index.h>
|
|
#include <faiss/IndexIDMap.h>
|
|
#include <faiss/IndexReplicas.h>
|
|
#include <faiss/IndexShards.h>
|
|
#include <vector>
|
|
|
|
namespace faiss {
|
|
|
|
/** splits input vectors in segments and assigns each segment to a sub-index
|
|
* used to distribute a MultiIndexQuantizer
|
|
*/
|
|
struct IndexSplitVectors : Index {
|
|
bool own_fields;
|
|
bool threaded;
|
|
std::vector<Index*> sub_indexes;
|
|
idx_t sum_d; /// sum of dimensions seen so far
|
|
|
|
explicit IndexSplitVectors(idx_t d, bool threaded = false);
|
|
|
|
void add_sub_index(Index*);
|
|
void sync_with_sub_indexes();
|
|
|
|
void add(idx_t n, const float* x) override;
|
|
|
|
void search(
|
|
idx_t n,
|
|
const float* x,
|
|
idx_t k,
|
|
float* distances,
|
|
idx_t* labels,
|
|
const SearchParameters* params = nullptr) const override;
|
|
|
|
void train(idx_t n, const float* x) override;
|
|
|
|
void reset() override;
|
|
|
|
~IndexSplitVectors() override;
|
|
};
|
|
|
|
/** index that returns random results.
|
|
* used mainly for time benchmarks
|
|
*/
|
|
struct IndexRandom : Index {
|
|
int64_t seed;
|
|
|
|
explicit IndexRandom(
|
|
idx_t d,
|
|
idx_t ntotal = 0,
|
|
int64_t seed = 1234,
|
|
MetricType mt = METRIC_L2);
|
|
|
|
void add(idx_t n, const float* x) override;
|
|
|
|
void search(
|
|
idx_t n,
|
|
const float* x,
|
|
idx_t k,
|
|
float* distances,
|
|
idx_t* labels,
|
|
const SearchParameters* params = nullptr) const override;
|
|
|
|
void reconstruct(idx_t key, float* recons) const override;
|
|
|
|
void reset() override;
|
|
|
|
~IndexRandom() override;
|
|
};
|
|
|
|
} // namespace faiss
|
|
|
|
#endif
|