结构体 faiss::AutoTuneCriterion

struct AutoTuneCriterion

评估标准。 返回[0,1]范围内的性能指标，值越高越好。

子类包括 faiss::IntersectionCriterion, faiss::OneRecallAtRCriterion

公共函数

AutoTuneCriterion(idx_t nq, idx_t nnn)

void set_groundtruth(int gt_nnn, const float *gt_D_in, const idx_t *gt_I_in)

初始化gt_D和gt_I向量。 必须在评估之前调用

参数:

• gt_D_in – 大小为 nq * gt_nnn

• gt_I_in – 大小为 nq * gt_nnn

virtual double evaluate(const float *D, const idx_t *I) const = 0

评估标准。

参数:

• D – 大小为 nq * nnn

• I – 大小为 nq * nnn

返回值:

标准，介于0和1之间。 值越大越好。

inline virtual ~AutoTuneCriterion()

公共成员

idx_t nq

此标准评估的查询数量

idx_t nnn

查询应请求的NN数量

idx_t gt_nnn

评估标准所需的GT NN数量

std::vector<float> gt_D

基本真值距离（大小为 nq * gt_nnn）

std::vector<idx_t> gt_I

基本真值索引（大小为 nq * gt_nnn）