liqian 1 年之前
父節點
當前提交
fb828a082c
共有 3 個文件被更改,包括 8 次插入2 次删除
  1. 7 1
      ad_model_run.sh
  2. 0 1
      ad_xgb_model_params_search.py
  3. 1 0
      ad_xgboost_predict.py

+ 7 - 1
ad_model_run.sh

@@ -41,4 +41,10 @@ then
   exit -1
 fi
 
-# 5. 离线预测
+# 5. 离线预测数据处理
+python ad_predict_video_data_process.py
+python ad_predict_user_data_process.py
+python ad_xgboost_predict_data_generate.py
+
+# 6. 离线预测
+python ad_xgboost_predict.py

+ 0 - 1
ad_xgb_model_params_search.py

@@ -40,7 +40,6 @@ grid = GridSearchCV(
     param_grid=parameters,
     scoring='roc_auc',
     n_jobs=4,
-    iid=False,
     cv=5)
 grid.fit(x_train, y_train)
 

+ 1 - 0
ad_xgboost_predict.py

@@ -23,4 +23,5 @@ df_0['y_1'] = y_pred_proba_1
 
 # 4. merge 结果
 res_df = pd.merge(df_0, df_1, how='left', on=['apptype', 'mid', 'videoid'])
+print(res_df.head())
 res_df['res_predict'] = res_df['y_0'] - res_df['y_1']