下载
中文
注册

train

API定义

void train(idx_t n, const float *x) override;

功能描述

对AscendIndexSQ执行训练量化器,继承AscendFaiss中的接口,并提供具体的实现。注意,执行add之前必须对Index进行train。

输入

idx_t n:训练集中特征向量的条数。

const float *x:特征向量数据。

输出

返回值

约束说明

  • 此处“n”的取值范围:0 < n < 1e9。
  • 此处指针“x”需要为非空指针,且长度应该为dims * n,否则可能出现越界读写的错误并引起程序崩溃。
  • 训练会统计的数据的分布,训练集比较小可能会影响查询精度。