Pārlūkot izejas kodu

feat:修改ros回归模型训练任务,添加样本权重

zhaohaipeng 1 mēnesi atpakaļ
vecāks
revīzija
77add54919

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

@@ -189,7 +189,7 @@ object recsys_01_ros_reg_weight_xgb_train {
         v(i + 1) = map.getOrDefault(features(i), 0.0d)
       }
       v(features.length + 1) = logKey
-      v(features.length + 2) = if (label == 0) 1 else 3.0 // 设置权重
+      v(features.length + 2) = if (label == 0) 1.1 else 3.0 // 设置权重
       Row(v: _*)
     })
   }