train

API定义

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

功能描述

对AscendIndexIVFSQT执行训练,继承AscendIndexIVFSQ中的相关接口并提供具体实现。

输入

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

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

输出

返回值

约束说明

  • 训练采用k-means进行聚类,训练集比较小可能会影响查询精度。
  • 此处“n”的取值范围:nlist ≤ n ≤ 7,000,000。
  • 此处指针“x”需要为非空指针,且长度应该为dimIn * n,否则可能出现越界读写的错误并引起程序崩溃。