Home Explore Help
Sign In
algorithm
/
recommend-model
15
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

feat:添加ros特征文件

zhaohaipeng 1 month ago
parent
ae44273796
commit
fcf1ea6403
1 changed files with 2 additions and 0 deletions
Split View Show Diff Stats
  1. 2 0
      recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/recsys_01_ros_multi_class_xgb_train.scala

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

@@ -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) {