Pārlūkot izejas kodu

feat:添加ros特征文件

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

+ 2 - 0
recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/recsys_01_ros_multi_class_xgb_train.scala

@@ -106,6 +106,8 @@ object recsys_01_ros_multi_class_xgb_train {
       .setNumWorkers(num_worker)
       .setSeed(2024)
       .setMinChildWeight(1)
+      .setNumClass(numClass)
+
     val model = xgbClassifier.fit(xgbInput)
 
     if (modelPath.nonEmpty && modelFile.nonEmpty) {