add printer

src/main.py

@@ -87,28 +87,63 @@ def main():
 
     @time_cost
     def test1():
+        time_1 = timeit.default_timer()
         user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
+        total_1 = (timeit.default_timer() - time_1)
+        print("join df cost {:.5f}s".format(total_1))
+
+        time_1 = timeit.default_timer()
         model_predict(user1, predict_fn)
+        total_1 = (timeit.default_timer() - time_1)
+        print("total prediction cost {:.5f}s".format(total_1))
 
     @time_cost
     def test2():
+        time_1 = timeit.default_timer()
         user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
+        total_1 = (timeit.default_timer() - time_1)
+        print("join df cost {:.5f}s".format(total_1))
+
+        time_1 = timeit.default_timer()
         model_predict(user1, predict_fn)
+        total_1 = (timeit.default_timer() - time_1)
+        print("total prediction cost {:.5f}s".format(total_1))
 
     @time_cost
     def test3():
+        time_1 = timeit.default_timer()
         user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
+        total_1 = (timeit.default_timer() - time_1)
+        print("join df cost {:.5f}s".format(total_1))
+
+        time_1 = timeit.default_timer()
         model_predict(user1, predict_fn)
+        total_1 = (timeit.default_timer() - time_1)
+        print("total prediction cost {:.5f}s".format(total_1))
 
     @time_cost
     def test4():
+        time_1 = timeit.default_timer()
         user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
+        total_1 = (timeit.default_timer() - time_1)
+        print("join df cost {:.5f}s".format(total_1))
+
+        time_1 = timeit.default_timer()
         model_predict(user1, predict_fn)
+        total_1 = (timeit.default_timer() - time_1)
+        print("total prediction cost {:.5f}s".format(total_1))
 
     @time_cost
     def test5():
+        time_1 = timeit.default_timer()
         user1 = join_device_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 300), redis_device_df, redis_diary_df)
+        total_1 = (timeit.default_timer() - time_1)
+        print("join df cost {:.5f}s".format(total_1))
+
+        time_1 = timeit.default_timer()
         model_predict(user1, predict_fn)
+        total_1 = (timeit.default_timer() - time_1)
+        print("total prediction cost {:.5f}s".format(total_1))
 
     test1()
     test2()
@@ -117,7 +152,7 @@ def main():
     test5()
 
     total_time = (time.time() - time_begin) / 60
-    print("cost {:.2f} mins at {}".format(total_time, datetime.now()))
+    print("total cost {:.2f} mins at {}".format(total_time, datetime.now()))
 
 
 if __name__ == "__main__":

src/models/esmm/model.py

+import timeit
+
 import tensorflow as tf
 from tensorflow import feature_column as fc
 from tensorflow.python.estimator.canned import head as head_lib
 from tensorflow.python.ops.losses import losses
-import timeit
 
 
 def build_deep_layer(net, params):
@@ -115,4 +116,5 @@ def model_predict(inputs, predict_fn):
     total_1 = (timeit.default_timer() - time_1)
     print("prediction cost {:.5f}s".format(total_1))
     # print(predictions)
+    # TODO get the diary ids
     return predictions