Commit f151f8d8 authored by 赵威's avatar 赵威

predict from redis

parent 4d681b0a
......@@ -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()))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment