|
@@ -105,8 +105,8 @@ object train_recsys_61_xgb_nor_20241209 {
|
|
|
val testDataSetTrans = vectorAssembler.transform(testDataSet).select("features", "label")
|
|
|
val predictions = model.transform(testDataSetTrans)
|
|
|
|
|
|
- println("recsys nor:columns:" + predictions.columns.mkString(",")) //[label, features, originalPrediction]
|
|
|
- val saveData = predictions.select("label", "originalPrediction").rdd
|
|
|
+ println("recsys nor:columns:" + predictions.columns.mkString(",")) //[label, features, prediction]
|
|
|
+ val saveData = predictions.select("label", "prediction").rdd
|
|
|
.map(r => {
|
|
|
(r.get(0), r.get(1)).productIterator.mkString("\t")
|
|
|
})
|
|
@@ -120,9 +120,9 @@ object train_recsys_61_xgb_nor_20241209 {
|
|
|
}
|
|
|
val evaluator = new RegressionEvaluator()
|
|
|
.setLabelCol("label")
|
|
|
- .setPredictionCol("originalPrediction")
|
|
|
+ .setPredictionCol("prediction")
|
|
|
.setMetricName("rmse")
|
|
|
- val rmse = evaluator.evaluate(predictions.select("label", "originalPrediction"))
|
|
|
+ val rmse = evaluator.evaluate(predictions.select("label", "prediction"))
|
|
|
println("recsys nor:rmse:" + rmse)
|
|
|
}
|
|
|
}
|