liqian hai 1 ano
pai
achega
75bb88a8b9
Modificáronse 1 ficheiros con 2 adicións e 0 borrados
  1. 2 0
      ad_xgboost_predict.py

+ 2 - 0
ad_xgboost_predict.py

@@ -33,6 +33,7 @@ print(res_df.head())
 
 # 5. to csv
 res_df.to_csv('./data/predict_data/predict_res.csv', index=False)
+print("to csv finished!")
 
 # 6. to redis
 for ind, row in res_df.iterrows():
@@ -42,4 +43,5 @@ for ind, row in res_df.iterrows():
     pre_res = row['res_predict']
     key = f"ad:xgb:predict:{app_type}:{mid}:{video_id}"
     redis_helper.set_data_to_redis(key_name=key, value=pre_res, expire_time=48*3600)
+print("to redis finished!")