我正在尝试获取GBTClassifier的ROC曲线。
一种方法是重用BinaryClassificationMetrics,但文档中给出的路径(https://spark.apache.org/docs/latest/mllib-evaluation-metrics.html)仅为ROC曲线提供了4个值,如:
[0.0|0.0]
[0.0|0.9285714285714286]
[1.0|1.0]
[1.0|1.0]
另一种方法是使用“概率”列而不是“预测”。但是,如果GBTClassifier
我没有它,此解决方案主要适用于RandomForestClassifier
。
How to plot ROC curve and precision-recall curve from BinaryClassificationMetrics
那么为任意分类器获得具有足够点的ROC曲线的一般/常用方法是什么?