|
@@ -40,7 +40,7 @@ object recsys_01_ros_multi_class_xgb_train {
|
|
|
val max_depth = param.getOrElse("max_depth", "5").toInt
|
|
|
val num_round = param.getOrElse("num_round", "100").toInt
|
|
|
val num_worker = param.getOrElse("num_worker", "20").toInt
|
|
|
- val func_object = param.getOrElse("func_object", "multi:softprob")
|
|
|
+ val func_object = param.getOrElse("func_object", "multi:softmax")
|
|
|
val func_metric = param.getOrElse("func_metric", "auc")
|
|
|
val repartition = param.getOrElse("repartition", "20").toInt
|
|
|
val numClass = param.getOrElse("numClass", "8").toInt
|