milvus/pyengine/engine/ingestion/serialize.py

19 lines
311 B
Python

import faiss
import numpy as np
def write_index(index, file_name):
faiss.write_index(index, file_name)
def read_index(file_name):
return faiss.read_index(file_name)
def to_array(vec):
return np.asarray(vec).astype('float32')
def to_int_array(vec):
return np.asarray(vec).astype('int64')