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')
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章