update path

src/main.py

@@ -62,9 +62,9 @@ def main():
     # save_path = model_export(model, all_features, model_export_path)
     # print("save to: " + save_path)
 
-    # save_path = "/home/gmuser/data/models/1595317247"
+    save_path = "/home/gmuser/data/models/1596012827"
 
-    save_path = str(Path("~/Desktop/models/1596012827").expanduser())
+    # save_path = str(Path("~/Desktop/models/1596012827").expanduser())
 
     # tf.saved_model.load
 
@@ -84,7 +84,8 @@ def main():
 
     for i in range(5):
         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)
+        res = model_predict_diary(random.sample(device_ids, 1)[0], random.sample(diary_ids, 200), device_dict, diary_dict, predict_fn)
+        print(res[:10])
         total_1 = (timeit.default_timer() - time_1)
         print("total prediction cost {:.5f}s".format(total_1), "\n")