Commit fa524138 authored by 赵威's avatar 赵威

try load model

parent 2df0646d
......@@ -13,11 +13,17 @@ data_dir = os.path.join(base_dir, "_data")
model_output_name = "w2v_model"
model_path = os.path.join(model_dir, model_output_name)
WORD2VEC_MODEL = word2vec.Word2Vec.load(model_path)
try:
WORD2VEC_MODEL = word2vec.Word2Vec.load(model_path)
except Exception as e:
print(e)
tracate_click_ids_model_name = "tractate_click_ids_item2vec_model"
tractate_click_ids_model_path = os.path.join(model_dir, tracate_click_ids_model_name)
TRACTATE_CLICK_IDS_MODEL = word2vec.Word2Vec.load(tractate_click_ids_model_path)
try:
TRACTATE_CLICK_IDS_MODEL = word2vec.Word2Vec.load(tractate_click_ids_model_path)
except Exception as e:
print(e)
class W2vSentences:
......
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