본문 바로가기
자연어처리(NLP)/IR(Information Retrieval)

[Faiss] index.search() return negative value, KeyError: -1

by Hyen4110 2022. 9. 2.
반응형

faiss 라이브러리를 정상적으로 사용하는데,

cpu_index.search(embedding, k) 결과 index값에 음수가 발생했다.

 

이후에 해당 document를 불러오는 과정에서 KeyErorr :-1

index값으로 음수를 반환하는 이유를 찾다가

embedding 값이 k보다 작음을 확인하였다. 

 

즉 검색 가능한 문서는 k 건인데,

원하는 검색건수가 >k여서

일단 index를 -1로 반환한것

 

POC mode로 작게만들어서 test 하던

embedding을 

원래 embedding으로 바꾼 뒤 정상적으로 작동함 확인하였다.

반응형

댓글