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