Commit 1e785aa6 authored by 赵威's avatar 赵威

update candidate size

parent cb72b020
......@@ -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):
time_begin = time.time()
diary_ids = user_portrait_tag3_get_candidate_unread_list(device_id, "diary")
diary_ids = user_portrait_tag3_get_candidate_unread_list(device_id, "diary", size=200)
print("diary_ids: " + str(len(diary_ids)))
res = model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_fn)
print("res: " + str(len(res)))
......@@ -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):
time_begin = time.time()
tractate_ids = user_portrait_tag3_get_candidate_unread_list(device_id, "tractate")
tractate_ids = user_portrait_tag3_get_candidate_unread_list(device_id, "tractate", size=200)
print("tractate_ids: " + str(len(tractate_ids)))
res = model_predict_tractate(device_id, tractate_ids, device_dict, tractate_dict, predict_fn)
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