scala train

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

@@ -100,9 +100,9 @@ object train_01_xgb_ad_20240808{
     val testDataSetTrans = vectorAssembler.transform(testDataSet).select("features","label")
     val predictions = model.transform(testDataSetTrans)
 
-    val saveData = predictions.select("label", "logKey", "rawPrediction", "probability").rdd
+    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/")) {