丁云鹏 8 tháng trước cách đây
mục cha
commit
c08ad0ec77

+ 1 - 1
recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/service/XGBoostService.java

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