|
@@ -101,7 +101,7 @@ public class XGBoostService {
|
|
Dataset<Row> selected = predictions.select("label", "rawPrediction");
|
|
Dataset<Row> selected = predictions.select("label", "rawPrediction");
|
|
BinaryClassificationEvaluator evaluator = new BinaryClassificationEvaluator()
|
|
BinaryClassificationEvaluator evaluator = new BinaryClassificationEvaluator()
|
|
.setLabelCol("label")
|
|
.setLabelCol("label")
|
|
- .setRawPredictionCol("rawPrediction")
|
|
|
|
|
|
+ .setRawPredictionCol("rawPrediction");
|
|
double auc = evaluator.evaluate(selected);
|
|
double auc = evaluator.evaluate(selected);
|
|
log.info("AUC: {}", auc);
|
|
log.info("AUC: {}", auc);
|
|
|
|
|