功能介绍
AscendIndexIVFast类,包含降维算法的三级检索AI算法,需要传入两个参数指明降维前后的维度信息,要求降维后维度能整除降维前的维度。
首先需要在GPU上训练AI聚类模型。
其次需要按照IVFSQT算子生成方式,生成三级检索所需算子。
此外,该类型也带有模糊聚类功能:入桶前,threshold参数控制底库冗余程度,搜索时建议使用batch size = 65536。
父主题: AscendIndexIVFast
AscendIndexIVFast类,包含降维算法的三级检索AI算法,需要传入两个参数指明降维前后的维度信息,要求降维后维度能整除降维前的维度。
首先需要在GPU上训练AI聚类模型。
其次需要按照IVFSQT算子生成方式,生成三级检索所需算子。
此外,该类型也带有模糊聚类功能:入桶前,threshold参数控制底库冗余程度,搜索时建议使用batch size = 65536。