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