赞
踩
提纲:
向量检索简单说是以向量为输入,希望在向量库中查询到最相近的目标向量。向量检索可以用于文本检索、文本匹配、图文检索等,主要是依靠深度学习模型的表示向量能够获得相比普通字符更大的表示空间,进而获得更精准的检索效果。
上图为使用向量检索工具构建检索模型的基本流程:
安装Faiss:pip install faiss-cpu
或GPU版本(默认使用CPU)pip install faiss-gpu
。
本例中,向量的维度为64,数据库规模为100000,待检索的查询词数量为10000,xb为数据库的对应向量库,维度100000*64,xq为查询词的向量表示,维度10000*64。
import numpy as np
d = 64 # dimension
nb = 100000 # database size
nq = 10000 # nb of queries
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。