瀏覽代碼

feat:添加ros特征文件

zhaohaipeng 1 月之前
父節點
當前提交
b6cb0ade68

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

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