【发布时间】:2023-02-10 18:08:53
【问题描述】:
我使用的是 titanic 数据集,所以它非常平衡(大约 60:40)并且 GaussianNB 模型(标准参数)的精度为 0.659。当我绘制 F1、精确度和召回率时,我发现了得分如此低的原因。
F1, precision and recall of GaussianNB
然后我计算了ROC-AUC,它是0.84。我花了几个小时试图了解这里发生的事情,但每个答案和博客文章主要是关于 ROC-AUC 在用于不平衡数据时的误导性。是什么让 ROC-AUC 如此之高,即使我们清楚地看到模型运行得不是很好?是因为召回率高吗?
LogisticRegression 的 ROC-AUC 分数:0.861
线性判别的ROC-AUC分数:0.859
KNeighbors 的 ROC-AUC 得分:0.855
SVC的ROC-AUC得分:0.836
GaussianProcess 的 ROC-AUC 分数:0.860
DecisionTree 的 ROC-AUC 得分:0.785
GaussianNB的ROC-AUC得分:0.840
ROC Curve for other models
【问题讨论】:
标签: machine-learning data-science roc auc precision-recall