serialize.py 305 字节
Newer Older
X
xj.lin 已提交
1
import faiss
X
xj.lin 已提交
2
import numpy as np
X
xj.lin 已提交
3 4 5 6 7

def write_index(index, file_name):
    faiss.write_index(index, file_name)

def read_index(file_name):
X
xj.lin 已提交
8 9 10
    return faiss.read_index(file_name)

def to_array(vec):
J
jinhai 已提交
11 12 13 14
    return np.asarray(vec).astype('float32')

def to_str_array(vec):
    return np.asarray(vec).astype('str')