昇腾社区首页
中文
注册

PretrainIndex

函数功能

预训练AscendKMeans使用的Index,节省重复创建Index的时间。

函数原型

APP_ERROR PretrainIndex(std::vector<float> &features, faiss::Index *&pretrainedIndex, bool addData);

参数说明

参数名

说明

features

输入的被执行特征聚类的向量,长度应该等于配置参数中的FeatureCount * Dim。

pretrainedIndex

输入待进行预训练的检索Index,需要设置为nullptr,执行完毕后返回一个非空的指针。

用户在使用完毕后,需通过显式delete释放该指针并置空。

addData

是否将输入的featureData添加进Index,如果设置为“false”,返回的pretrainedIndex中的底库特征数量为空,如果设置为“true”,那么执行完预训练之后也会将feature中的数据添加进pretrainedIndex中。

返回参数说明

数据结构

说明

APP_ERROR

返回0表示预训练完成。