predict from redis

src/main.py

@@ -52,31 +52,31 @@ def main():
     save_path = "/home/gmuser/data/models/1595317247"
     predict_fn = tf.contrib.predictor.from_saved_model(save_path)
 
-    for i in range(10):
-        test_300 = test_df.sample(300)
-        model_predict(test_300, predict_fn)
+    # for i in range(10):
+    #     test_300 = test_df.sample(300)
+    #     model_predict(test_300, predict_fn)
 
-    # print("==============================")
-    # device_id = "861601036552944"
+    print("==============================")
+    device_id = "861601036552944"
     # diary_ids = [
     #     "16195283", "16838351", "17161073", "17297878", "17307484", "17396235", "16418737", "16995481", "17312201", "12237988"
     # ]
 
-    # df = get_device_df_from_redis()
-    # df2 = get_diary_df_from_redis()
-    # redis_device_df = device_feature_engineering(df)
-    # redis_diary_df = diary_feature_engineering(df2, from_redis=True)
-    # diary_ids = list(redis_diary_df["card_id"].values)[:300]
-
-    # time_1 = timeit.default_timer()
-    # res = join_device_diary(device_id, diary_ids, redis_device_df, redis_diary_df)
-    # print(len(res))
-    # print(res.sample(1), "\n")
-    # print(res.sample(1))
-
-    # # model_predict(res, predict_fn)
-    # total_1 = (timeit.default_timer() - time_1)
-    # print("prediction total cost {:.5f}s".format(total_1))
+    df = get_device_df_from_redis()
+    df2 = get_diary_df_from_redis()
+    redis_device_df = device_feature_engineering(df)
+    redis_diary_df = diary_feature_engineering(df2, from_redis=True)
+    diary_ids = list(redis_diary_df["card_id"].values)[:300]
+
+    time_1 = timeit.default_timer()
+    res = join_device_diary(device_id, diary_ids, redis_device_df, redis_diary_df)
+    print(len(res))
+    print(res.sample(1), "\n")
+    print(res.sample(1))
+
+    model_predict(res, predict_fn)
+    total_1 = (timeit.default_timer() - time_1)
+    print("prediction total cost {:.5f}s".format(total_1))
 
     total_time = (time.time() - time_begin) / 60
     print("cost {:.2f} mins at {}".format(total_time, datetime.now()))