test

src/main.py

@@ -74,17 +74,17 @@ def main():
     #     "16195283", "16838351", "17161073", "17297878", "17307484", "17396235", "16418737", "16995481", "17312201", "12237988"
     # ]
 
-    # device_dict = get_device_dict_from_redis()
-    # diary_dict = get_diary_dict_from_redis()
+    device_dict = get_device_dict_from_redis()
+    diary_dict = get_diary_dict_from_redis()
 
-    # device_ids = list(device_dict.keys())[:20]
-    # diary_ids = list(diary_dict.keys())
+    device_ids = list(device_dict.keys())[:20]
+    diary_ids = list(diary_dict.keys())
 
-    # for i in range(2):
-    #     time_1 = timeit.default_timer()
-    #     model_predict_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 200), device_dict, diary_dict, predict_fn)
-    #     total_1 = (timeit.default_timer() - time_1)
-    #     print("total prediction cost {:.5f}s".format(total_1), "\n")
+    for i in range(2):
+        time_1 = timeit.default_timer()
+        model_predict_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 200), device_dict, diary_dict, predict_fn)
+        total_1 = (timeit.default_timer() - time_1)
+        print("total prediction cost {:.5f}s".format(total_1), "\n")
 
     total_time = (time.time() - time_begin) / 60
     print("total cost {:.2f} mins at {}".format(total_time, datetime.now()))