try predict diary

src/main_portrait.py

@@ -14,7 +14,7 @@ from utils.portrait import (get_user_portrait_tag3_read_v2, user_portrait_tag3_g
                             user_portrait_tag3_get_candidate_unread_list, user_portrait_tag3_write_ctcvr_data)
 
 
-def user_portrait_scan_info(device_dict, diary_dict, predict_fn, tail_number):
+def user_portrait_scan_info(device_dict, diary_dict, tractate_dict, diary_predict_fn, tractate_predict_fn, tail_number):
     try:
         round = 0
         all_count = 0
@@ -32,7 +32,8 @@ def user_portrait_scan_info(device_dict, diary_dict, predict_fn, tail_number):
                     if (user_portrait_tag3_get_candidate_dict(device_id, "diary")):
                         all_count += 1
                         print(str(all_count) + ": " + device_id)
-                        offline_predict_diary(device_id, device_dict, diary_dict, predict_fn)
+                        offline_predict_diary(device_id, device_dict, diary_dict, diary_predict_fn)
+                        # offline_predict_tractate(device_dict, device_dict, tractate_dict, tractate_predict_fn)
 
         print("all count: " + str(all_count))
         print("scan done " + str(datetime.datetime.now()))
@@ -54,6 +55,20 @@ def offline_predict_diary(device_id, device_dict, diary_dict, predict_fn):
     print("total cost {:.5f}s\n".format(total_time))
 
 
+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")
+    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)))
+    # print(res[:10])
+    user_portrait_tag3_write_ctcvr_data(device_id, "tractate", res[:500])
+
+    total_time = (time.time() - time_begin)
+    print("total cost {:.5f}s\n".format(total_time))
+
+
 def main():
     tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.INFO)
 
@@ -76,7 +91,7 @@ def main():
     # print(res[:10])
 
     tail_number = sys.argv[1]  # "c", "d", "e", "f"
-    user_portrait_scan_info(device_dict, diary_dict, diary_predict_fn, tail_number)
+    user_portrait_scan_info(device_dict, diary_dict, tractate_dict, diary_predict_fn, tractate_predict_fn, tail_number)
 
 
 if __name__ == "__main__":