下载
中文
注册

train

API定义

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

功能描述

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

输入

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

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

输出

返回值

约束说明

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