feat:修改XGB训练脚本

recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/train_01_xgb_ad_20240808.scala → recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/recsys_01_xgb_train.scala

@@ -15,7 +15,7 @@ import java.util
 import scala.collection.mutable.ArrayBuffer
 import scala.io.Source
 
-object train_01_xgb_ad_20240808 {
+object recsys_01_xgb_train {
   def main(args: Array[String]): Unit = {
     val spark = SparkSession
       .builder()
@@ -107,7 +107,7 @@ object train_01_xgb_ad_20240808 {
     println("zhangbo:columns:" + predictions.columns.mkString(","))
     val saveData = predictions.select("label", "rawPrediction", "probability").rdd
       .map(r => {
-        (r.get(0), r.get(1), r.get(2), r.get(3)).productIterator.mkString("\t")
+        (r.get(0), r.get(1), r.get(2)).productIterator.mkString("\t")
       })
     val hdfsPath = savePath
     if (hdfsPath.nonEmpty && hdfsPath.startsWith("/dw/recommend/model/")) {