Skip to content

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

get resurt

parent d7ba7264
Show whitespace changes
Inline Side-by-side
Showing with 22 additions and 5 deletions
src/diary_cover_similarity.py
View file @ cbbc2fb0
...@@ -18,6 +18,14 @@ def save_diary_image_info(save_file, face_to_vec_f): ...@@ -18,6 +18,14 @@ def save_diary_image_info(save_file, face_to_vec_f):
"term": { "term": {
"is_online": True "is_online": True
} }
}, {
"term": {
"has_cover": True
}
}, {
"term": {
"is_sink": False
}
}, { }, {
"term": { "term": {
"has_before_cover": True "has_before_cover": True
...@@ -26,6 +34,14 @@ def save_diary_image_info(save_file, face_to_vec_f): ...@@ -26,6 +34,14 @@ def save_diary_image_info(save_file, face_to_vec_f):
"term": { "term": {
"has_after_cover": True "has_after_cover": True
} }
}, {
"terms": {
"content_level": [6, 5, 4, 3.5, 3]
}
}, {
"term": {
"content_simi_bol_show": 0
}
}, { }, {
"exists": { "exists": {
"field": "before_cover_url" "field": "before_cover_url"
...@@ -122,11 +138,12 @@ def main(): ...@@ -122,11 +138,12 @@ def main():
] ]
feature = np.array(a).astype("float32") feature = np.array(a).astype("float32")
index = faiss.read_index(faiss_index_path) index = faiss.read_index(faiss_index_path)
D, I = index.search(np.array[feature], 5) D, I = index.search(np.array([feature]), 5)
print(D) ids = I.flat
print(I) scores = D.flat
ids = list(I.flat) res = list(zip(ids, scores))
print(ids) res.sort(key = lambda x: x[1], reverse=True)
print(res)
if __name__ == "__main__": if __name__ == "__main__":
......
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