Igo*_*tov 5 machine-learning roc apache-spark apache-spark-mllib
我正在尝试获取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]
Run Code Online (Sandbox Code Playgroud)
另一种方法是使用"概率"列而不是"预测".但是,如果GBTClassifier我没有它,这个解决方案主要适用于RandomForestClassifier.
如何从BinaryClassificationMetrics绘制ROC曲线和精确回忆曲线
那么为任意分类器获得具有足够点的ROC曲线的一般/常用方法是什么?
| 归档时间: |
|
| 查看次数: |
935 次 |
| 最近记录: |