자연어처리(NLP)/IR(Information Retrieval)
[Faiss] index.search() return negative value, KeyError: -1
Hyen4110
2022. 9. 2. 16:31
faiss 라이브러리를 정상적으로 사용하는데,
cpu_index.search(embedding, k) 결과 index값에 음수가 발생했다.
이후에 해당 document를 불러오는 과정에서 KeyErorr :-1
index값으로 음수를 반환하는 이유를 찾다가
embedding 값이 k보다 작음을 확인하였다.
즉 검색 가능한 문서는 k 건인데,
원하는 검색건수가 >k여서
일단 index를 -1로 반환한것
POC mode로 작게만들어서 test 하던
embedding을
원래 embedding으로 바꾼 뒤 정상적으로 작동함 확인하였다.