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表示预训练完成。 |
父主题: KMeans离线index预训练