Я пытаюсь получить кривую ROC для GBTClassifier.
Один из способов — повторно использовать BinaryClassificationMetrics, однако путь, указанный в документации (https://spark.apache.org/docs/latest/mllib-evaluation-metrics.html) предоставляет только 4 значения для кривой ROC, например:
[0.0|0.0]
[0.0|0.9285714285714286]
[1.0|1.0]
[1.0|1.0]
Другой способ — использовать столбец «вероятность» вместо «прогноз». Однако в случае GBTClassifier
у меня его нет, и это решение работает в основном для RandomForestClassifier
.
Как построить кривую ROC и кривую точного отзыва из BinaryClassificationMetrics
Итак, каков общий/общий способ получить кривую ROC с достаточным количеством точек для произвольного классификатора?