try get result

src/main_portrait.py

@@ -9,7 +9,7 @@ from utils.cache import redis_client2
 from utils.portrait import (user_portrait_tag3_get_candidate_unread_list, user_portrait_tag3_write_ctcvr_data)
 
 
-def user_portrait_scan_info():
+def user_portrait_scan_info(device_dict, diary_dict, predict_fn):
     try:
         round = 0
         all_count = 0
@@ -22,8 +22,9 @@ def user_portrait_scan_info():
             for key in results:
                 key = str(key, "utf-8")
                 device_id = key.split(":")[-1]
-                print(device_id)
-                print(user_portrait_tag3_get_candidate_unread_list(device_id, "diary")[:10])
+
+                offline_predict(device_id, device_dict, diary_dict, predict_fn)
+
                 all_count += 1
 
         print("all count: " + str(all_count))
@@ -32,26 +33,30 @@ def user_portrait_scan_info():
         print(e)
 
 
+def offline_predict(device_id, device_dict, diary_dict, predict_fn):
+    diary_ids = user_portrait_tag3_get_candidate_unread_list(device_id, "diary")
+    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)
+
+
 def main():
-    time_begin = time.time()
     tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.INFO)
 
     device_dict = get_device_dict_from_redis()
     diary_dict = get_diary_dict_from_redis()
 
-    device_id = "androidid_a25a1129c0b38f7b"
+    # device_id = "androidid_a25a1129c0b38f7b"
     save_path = "/home/gmuser/data/models/1595501144"
     predict_fn = tf.contrib.predictor.from_saved_model(save_path)
 
-    diary_ids = user_portrait_tag3_get_candidate_unread_list(device_id, "diary")
-    res = model_predict_diary(device_id, diary_ids, device_dict, diary_dict, predict_fn)
-    print(res)
-    user_portrait_tag3_write_ctcvr_data(device_id, "diary", res)
-    # user_portrait_scan_info()
-
-    total_time = (time.time() - time_begin) / 60
-    print("total cost {:.2f} mins at {}".format(total_time, datetime.datetime.now()))
+    user_portrait_scan_info(device_dict, diary_dict, predict_fn)
 
 
 if __name__ == "__main__":
+    time_begin = time.time()
+
     main()
+
+    total_time = (time.time() - time_begin) / 60
+    print("total cost {:.2f} mins at {}".format(total_time, datetime.datetime.now()))