Skip to content

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

try get item2vec

parent 4241dc50
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
word_vector/api.py
View file @ 5f51724f
import multiprocessing import multiprocessing
import os import os
import sys import sys
import time
import traceback import traceback
sys.path.append(os.path.realpath(".")) sys.path.append(os.path.realpath("."))
...@@ -89,8 +88,10 @@ def projects_item2vec(score_limit=5): ...@@ -89,8 +88,10 @@ def projects_item2vec(score_limit=5):
def clicked_tractate_ids_item2vec_model(id, n=5): def clicked_tractate_ids_item2vec_model(id, n=5):
try: try:
return TRACTATE_CLICK_IDS_MODEL.wv.most_similar(id, topn=n) return TRACTATE_CLICK_IDS_MODEL.wv.most_similar(id, topn=n)
except KeyError as e:
send_msg_to_dingtalk("tractate_item2vec: " + str(e))
except Exception as e: except Exception as e:
# send_msg_to_dingtalk(str(traceback.format_exc())) send_msg_to_dingtalk(str(traceback.format_exc()))
return [] return []
......
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