try predict

src/main.py

@@ -27,7 +27,7 @@ def time_cost(func):
         ret = func(*args, **kwargs)
 
         total_1 = (timeit.default_timer() - time_1)
-        print("prediction total cost {:.5f}s".format(total_1))
+        print("cost {:.5f}s".format(total_1))
         return ret
 
     return wrapper
@@ -89,18 +89,31 @@ def main():
         user1 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
         model_predict(user1, predict_fn)
 
-    test1()
+    @time_cost
+    def test2():
+        user1 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
+        model_predict(user1, predict_fn)
+
+    @time_cost
+    def test3():
+        user1 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
+        model_predict(user1, predict_fn)
 
-    # time_1 = timeit.default_timer()
-    # user1 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
-    # model_predict(user1, predict_fn)
-    # total_1 = (timeit.default_timer() - time_1)
-    # print("prediction total cost {:.5f}s".format(total_1))
+    @time_cost
+    def test4():
+        user1 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
+        model_predict(user1, predict_fn)
+
+    @time_cost
+    def test5():
+        user1 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
+        model_predict(user1, predict_fn)
 
-    # user2 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
-    # user3 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
-    # user4 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
-    # user5 = join_device_diary(device_id, random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
+    test1()
+    test2()
+    test3()
+    test4()
+    test5()
 
     total_time = (time.time() - time_begin) / 60
     print("cost {:.2f} mins at {}".format(total_time, datetime.now()))