debug

ad_xgboost_predict.py

@@ -34,13 +34,17 @@ def xgboost_predict(model, app_type, mid, video_id, abtest_id, ab_test_code):
     ad_feature_0 = user_feature + video_feature + [0]
     # ad_0_predict = model.predict_proba(np.array([ad_feature_0]))
     # ad_0_predict = ad_0_predict[0][1]
-    ad_0_predict = 0.7
+    # ad_0_predict = 0.7
 
     # 4. 拼接出广告时的特征 & 预测
     ad_feature_1 = user_feature + video_feature + [1]
     # ad_1_predict = model.predict_proba(np.array([ad_feature_1]))
     # ad_1_predict = ad_1_predict[0][1]
-    ad_1_predict = 0.6
+    # ad_1_predict = 0.6
+
+    # 预测
+    predict = model.predict_proba(np.array([ad_feature_0, ad_feature_1]))
+    ad_0_predict, ad_1_predict = predict[0][1], predict[1][1]
 
     # 5. 作差
     predict_res = ad_0_predict - ad_1_predict