add printer

src/main_portrait.py

@@ -35,6 +35,7 @@ def user_portrait_scan_info(device_dict, diary_dict, predict_fn):
 
 def offline_predict(device_id, device_dict, diary_dict, predict_fn):
     diary_ids = user_portrait_tag3_get_candidate_unread_list(device_id, "diary")
+    print(diary_ids)
     res = model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_fn)
     print(res[:10])
     # user_portrait_tag3_write_ctcvr_data(device_id, "diary", res)
@@ -47,11 +48,13 @@ def main():
     diary_dict = get_diary_dict_from_redis()
     print(len(device_dict), len(diary_dict))
 
-    # device_id = "androidid_a25a1129c0b38f7b"
     save_path = "/home/gmuser/data/models/1595501144"
     predict_fn = tf.contrib.predictor.from_saved_model(save_path)
 
-    user_portrait_scan_info(device_dict, diary_dict, predict_fn)
+    device_id = "androidid_a25a1129c0b38f7b"
+    offline_predict(device_id, device_dict, diary_dict, predict_fn)
+
+    # user_portrait_scan_info(device_dict, diary_dict, predict_fn)
 
 
 if __name__ == "__main__":