Skip to content

S
strategy_embedding
Commit 5478c0c1 authored by 赵威's avatar 赵威
Browse files
Options

get info

parent 4ab16ab4
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/diary_cover_similarity.py
View file @ 5478c0c1
...@@ -68,7 +68,9 @@ def save_faiss_index(load_file, save_path): ...@@ -68,7 +68,9 @@ def save_faiss_index(load_file, save_path):
ids_np = np.array(ids).astype("float32") ids_np = np.array(ids).astype("float32")
features_np = np.array(features).astype("int") features_np = np.array(features).astype("int")
print(ids_np) print(ids_np)
print(ids.dtype)
print(features_np) print(features_np)
print(features_np.dtype)
index = faiss.IndexHNSW(128, 32) index = faiss.IndexHNSW(128, 32)
print(index.is_trained) print(index.is_trained)
index2 = faiss.IndexIDMap(index) index2 = faiss.IndexIDMap(index)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment