浏览代码

feat:添加分桶脚本

zhaohaipeng 2 月之前
父节点
当前提交
e7e04b7b76

+ 1 - 1
recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/train_01_xgb_ad_20240808.scala

@@ -158,7 +158,7 @@ object train_01_xgb_ad_20240808 {
         map.put(fv(0), NumberUtils.toDouble(fv(1), 0.0))
       }
 
-      val v: Array[Any] = new Array[Any](features.length + 2)
+      val v: Array[Any] = new Array[Any](features.length + 1)
       v(0) = label
       for (i <- 0 until features.length) {
         v(i + 1) = map.getOrDefault(features(i), 0.0d)