Переглянути джерело

feat:添加ros回归模型

zhaohaipeng 1 місяць тому
батько
коміт
9d5b74f4a7

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

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