return empyt data

186fcd66 · 赵威 · fa524138 · 186fcd66
Commit 186fcd66 authored Oct 28, 2020 by 赵威
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

word_to_vec.py word_vector/word_to_vec.py +4 -1

No files found.
--- a/word_vector/word_to_vec.py
+++ b/word_vector/word_to_vec.py
@@ -103,7 +103,10 @@ def save_clicked_tractate_ids_item2vec():

 @bind("strategy_embedding/word_vector/tractate_item2vec")
 def clicked_tractate_ids_item2vec_model(id, n=5):
-    return TRACTATE_CLICK_IDS_MODEL.wv.most_similar(id, topn=n)
+    try:
+        return TRACTATE_CLICK_IDS_MODEL.wv.most_similar(id, topn=n)
+    except Exception as e:
+        return []


 if __name__ == "__main__":