[Python] Faiss, 효율적인 유사도 검색 엔진
·
Python
Faiss(Facebook AI Similarity Search)는 페이스북에서 만든 유사도 검색 라이브러리입니다. GPU를 사용할 수 있고 C++ 기반이기 때문에 sklearn보다 빠르다는 장점이 있습니다. 0. 설치 pip install faiss-gpu or pip install faiss-cpu 1. 벡터 생성 후 index 구축 ※ faiss는 index라는 객체를 사용합니다. 쉽게 말해서 db라고 생각하면 될 것 같습니다. import numpy as np import faiss # 5차원 벡터 10000개 생성 db_vector = np.array(np.random.random((10000,5)),np.float32) # 유사도 찾을 벡터 query_vector = np.array(np.ra..