结构体 faiss::OneRecallAtRCriterion

struct OneRecallAtRCriterion : public faiss::AutoTuneCriterion

公共函数

OneRecallAtRCriterion(idx_t nq, idx_t R)

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

评估标准。

参数:

• D – 大小 nq * nnn

• I – 大小 nq * nnn

返回值:

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

inline ~OneRecallAtRCriterion() override

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

公共成员

idx_t R

idx_t nq

此标准评估的查询数量

idx_t nnn

查询应请求的 NN 的数量

idx_t gt_nnn

评估标准所需的 GT NN 的数量

std::vector<float> gt_D

Ground-truth 距离（大小 nq * gt_nnn）

std::vector<idx_t> gt_I

Ground-truth 索引（大小 nq * gt_nnn）