Parcourir la source

feat:修改ros多分类模型

zhaohaipeng il y a 1 mois
Parent
commit
b745cf3d86

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

@@ -30,6 +30,8 @@ object recsys_01_ros_multi_class_xgb_train {
       .getOrCreate()
     val sc = spark.sparkContext
 
+    import spark.implicits._
+
     val param = ParamUtils.parseArgs(args)
     val featureFile = param.getOrElse("featureFile", "20250306_ros_feature_232.txt")
     val trainPath = param.getOrElse("trainPath", "/dw/recommend/model/43_recsys_ros_data_bucket/20250301")