Data Science/Python
[Python] Faiss 활용한 Vector 유사도 측정 및 검색
faiss : 페이스북에서 C++로 제작한 Vector 자료 구조를 색인화하고 유사성을 검색하는 모듈이다. CPU 기반의 모듈(pip install faiss-cpu )과 GPU (pip install faiss-gpu ) 모듈이 각각이 있어 환경에 맞는 모듈을 설치한다. # pip install faiss-cpu import faiss def create_index_kmean(vector, size, saved_file=None): dimension = 768 index_obj = faiss.IndexIDMap( faiss.IndexFlatIP(dimension) ) xidx = np.array( range(0, size), dtype='int64' ) index_obj.add_with_ids(vect..
2022. 9. 17. 06:57
최근댓글