update timer

requirements.txt

 tensorflow-gpu==1.15.2
 keras==2.3.1
+pandas==1.0.5
 scikit-learn==0.23.1
 
 redis==2.10.5

src/main.py

 import os
+import shutil
 import time
+import timeit
 from datetime import datetime
 from pathlib import Path
-import shutil
 
 import tensorflow as tf
 from sklearn.model_selection import train_test_split
@@ -46,10 +47,10 @@ def main():
     # print(next(iter(predictions)))
 
     test_300 = test_df.sample(300)
-    time_1 = time.time()
+    time_1 = timeit.default_timer()
     model_predict(test_300, save_path)
-    total_1 = (time.time() - time_1)
-    print("prediction cost {:.5f} s at {}".format(total_1, datetime.now()))
+    total_1 = (timeit.default_timer() - time_1)
+    print("prediction cost {:.5f} s".format(total_1))
 
     total_time = (time.time() - time_begin) / 60
     print("cost {:.2f} mins at {}".format(total_time, datetime.now()))