get res

src/main_portrait.py

@@ -43,7 +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))
+    print("res: " + len(res) + " " + str(res[:5]))
     user_portrait_tag3_write_ctcvr_data(device_id, "diary", res[:500])
 
     total_time = (time.time() - time_begin)
@@ -61,11 +61,11 @@ def main():
     predict_fn = tf.contrib.predictor.from_saved_model(save_path)
 
     device_id = "androidid_a25a1129c0b38f7b"
-    # offline_predict(device_id, device_dict, diary_dict, predict_fn)
+    offline_predict(device_id, device_dict, diary_dict, predict_fn)
 
-    res = user_portrait_tag3_get_candidate_unread_list(device_id, "diary")
-    print(len(res))
-    print(res[:10])
+    # res = user_portrait_tag3_get_candidate_unread_list(device_id, "diary")
+    # print(len(res))
+    # print(res[:10])
 
     # user_portrait_scan_info(device_dict, diary_dict, predict_fn)