Skip to content

G
gm_strategy_cvr
Commit a6d6d06e authored by 赵威's avatar 赵威
Browse files
Options

change prediction number

parent a8a59f31
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/main_portrait.py
View file @ a6d6d06e
...@@ -45,7 +45,7 @@ def user_portrait_scan_info(device_dict, diary_dict, tractate_dict, diary_predic ...@@ -45,7 +45,7 @@ def user_portrait_scan_info(device_dict, diary_dict, tractate_dict, diary_predic
def offline_predict_diary(device_id, device_dict, diary_dict, predict_fn): def offline_predict_diary(device_id, device_dict, diary_dict, predict_fn):
time_begin = time.time() time_begin = time.time()
diary_ids = user_portrait_tag3_get_candidate_unread_list(device_id, "diary", size=200) diary_ids = user_portrait_tag3_get_candidate_unread_list(device_id, "diary", size=300)
print("diary_ids: " + str(len(diary_ids))) print("diary_ids: " + str(len(diary_ids)))
res = model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_fn) res = model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_fn)
print("res: " + str(len(res))) print("res: " + str(len(res)))
...@@ -59,7 +59,7 @@ def offline_predict_diary(device_id, device_dict, diary_dict, predict_fn): ...@@ -59,7 +59,7 @@ def offline_predict_diary(device_id, device_dict, diary_dict, predict_fn):
def offline_predict_tractate(device_id, device_dict, tractate_dict, predict_fn): def offline_predict_tractate(device_id, device_dict, tractate_dict, predict_fn):
time_begin = time.time() time_begin = time.time()
tractate_ids = user_portrait_tag3_get_candidate_unread_list(device_id, "tractate", size=200) tractate_ids = user_portrait_tag3_get_candidate_unread_list(device_id, "tractate", size=300)
print("tractate_ids: " + str(len(tractate_ids))) print("tractate_ids: " + str(len(tractate_ids)))
res = model_predict_tractate(device_id, tractate_ids, device_dict, tractate_dict, predict_fn) res = model_predict_tractate(device_id, tractate_ids, device_dict, tractate_dict, predict_fn)
print("res: " + str(len(res))) print("res: " + str(len(res)))
......
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