get result

src/main_portrait.py

@@ -43,6 +43,7 @@ def offline_predict(device_id, device_dict, diary_dict, predict_fn):
     print("diary_ids: " + str(len(diary_ids)))
     diary_ids = random.sample(diary_ids, 500)
     res = model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_fn)
+    print("res: " + len(res))
     user_portrait_tag3_write_ctcvr_data(device_id, "diary", res[:500])
 
     total_time = (time.time() - time_begin)

src/utils/portrait.py

@@ -144,8 +144,8 @@ def user_portrait_tag3_get_candidate_unread_list(device_id, content_type, size=5
     if diary_dict:
         # lol = remove_read_with_size(diary_dict.values(), "diary", read_list, size)
         lol = diary_dict.values()
-        queue = merge_lists(*lol)
-        get_n_unread_data_from_ids(queue, size, read_list)
+        lists = merge_lists(*lol)
+        queue = get_n_unread_data_from_ids(lists, size, read_list)
     return queue