Skip to content

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

add printer

parent 5545458c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diary_cover_similarity.py
View file @ ce0e5827
...@@ -84,7 +84,7 @@ def save_faiss_index(load_file, save_path): ...@@ -84,7 +84,7 @@ def save_faiss_index(load_file, save_path):
tmp = line.split("\t") tmp = line.split("\t")
ids.append(tmp[0]) ids.append(tmp[0])
features.append(np.array(json.loads(tmp[1]))) features.append(np.array(json.loads(tmp[1])))
print(count + " " + tmp[0]) print("{} {}".format(count, tmp[0]))
ids_np = np.array(ids).astype("int") ids_np = np.array(ids).astype("int")
features_np = np.array(features).astype("float32") features_np = np.array(features).astype("float32")
......
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