丁云鹏 8 місяців тому
батько
коміт
2948a42836

+ 23 - 14
recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/service/XGBoostService.java

@@ -109,20 +109,29 @@ public class XGBoostService {
 
     private static Dataset<Row> dataset(String path) {
         String[] features = {"cpa",
-                "b2_12h_ctr",
-                "b2_12h_ctcvr",
-                "b2_12h_cvr",
-                "b2_12h_conver",
-                "b2_12h_click",
-                "b2_12h_conver*log(view)",
-                "b2_12h_conver*ctcvr",
-                "b2_7d_ctr",
-                "b2_7d_ctcvr",
-                "b2_7d_cvr",
-                "b2_7d_conver",
-                "b2_7d_click",
-                "b2_7d_conver*log(view)",
-                "b2_7d_conver*ctcvr"
+                "b2_1h_ctr",
+                "b2_1h_ctcvr",
+                "b2_1h_cvr",
+                "b2_1h_conver",
+                "b2_1h_click",
+                "b2_1h_conver*log(view)",
+                "b2_1h_conver*ctcvr",
+                "b2_2h_ctr",
+                "b2_2h_ctcvr",
+                "b2_2h_cvr",
+                "b2_2h_conver",
+                "b2_2h_click",
+                "b2_2h_conver*log(view)",
+                "b2_2h_conver*ctcvr",
+                "b2_3h_ctr",
+                "b2_3h_ctcvr",
+                "b2_3h_cvr",
+                "b2_3h_conver",
+                "b2_3h_click",
+                "b2_3h_conver*log(view)",
+                "b2_3h_conver*ctcvr",
+                "b2_6h_ctr",
+                "b2_6h_ctcvr"
         };