|
|
@@ -30,8 +30,6 @@ 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")
|
|
|
@@ -149,6 +147,7 @@ object recsys_01_ros_multi_class_xgb_train {
|
|
|
}
|
|
|
|
|
|
|
|
|
+ import spark.implicits._
|
|
|
// 计算 Multiclass Log Loss
|
|
|
val logLossDF = predictions.withColumn("log_loss",
|
|
|
udf((label: Double, probVec: Vector) => {
|
