milvus/pyengine/engine/ingestion/serialize.py

15 lines
305 B
Python
Raw Normal View History

import faiss
2019-03-25 08:11:15 +00:00
import numpy as np
def write_index(index, file_name):
faiss.write_index(index, file_name)
def read_index(file_name):
2019-03-25 08:11:15 +00:00
return faiss.read_index(file_name)
def to_array(vec):
2019-03-26 13:28:20 +00:00
return np.asarray(vec).astype('float32')
def to_str_array(vec):
return np.asarray(vec).astype('str')