提交 cdc28432 编写于 作者: X xj.lin

1. organize project folder structure 2. add faiss_wrapper interface

上级 3818ea20
import faiss
import numpy as np
class FaissSearch():
def __init__(self, index, id_to_vector_map):
pass
# def search_by_ids(self, id_list, k):
# pass
def search_by_vectors(self, vector_list, k):
# return both id and vector
pass
def __search__(self, id_list, vector_list, k):
pass
class FaissIndex():
def build_index(self, vector_list, dimension):
# return index
pass
# def build_index_cpu(self):
# pass
# def build_index_gpu(self):
# pass
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册