绘制ROC曲线的常用方法

时间:2017-02-16 15:07:36

标签: apache-spark machine-learning apache-spark-mllib roc

我正在尝试获取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曲线的一般/常用方法是什么?

0 个答案:

没有答案