faiss遇到的坑

1、

TypeError: in method ‘IndexFlat_add’, argument 3 of type ‘float const*’

数组不能是float64，要是float32

查看数据类型

arr.dtype

arr.astype(‘float32’)

 

2、

n, d = x.shape

ValueError: not enough values to unpack (expected 2, got 1)

x.shape返回的只有一个数据，而左边有两个，所以报错

原因是我输入的数据问题

arr[5, :].shape, 结果(512,)

arr[:, 5].shape ，结果(500,)

以上结果发现，arr[5, :],arr[:, 1]只返回行向量，即便是列向量，也转为了行向量

且返回的是第五行，而不是返回前五行，前片的问题

 

要加切片

arr[:5, :]是5行，n列

或者arr[:5]

 

3、

self.add_with_ids_c(n, swig_ptr(x), swig_ptr(ids))

ValueError: did not recognize array type

 

换数据格式，ids数据格式换成int，我的就可以了

ids = ids.astype('int')