如何计算分类模型不同阈值的TPR和FPR?

数据挖掘 机器学习 爪哇
2021-09-25 00:54:41

我已经建立了一个分类模型来预测二元类。我可以计算精度、召回率和 F1-Score。现在,我想生成 ROC 以更好地了解我的分类模型的分类性能。我不知道如何计算不同阈值的 TPR 和 FPR。

2个回答

要计算不同阈值的 TPR 和 FPR,您可以按照以下步骤操作:

  1. 首先计算每个类的预测概率,而不是类预测。

  2. 根据正类的概率值对测试用例进行排序(假设二元类为正类和负类)。

  3. 然后在概率分数上设置不同的截止/阈值并计算 TPR=TP(TP + FP)FPR=FP(FP + TN)对于每个阈值。

上面的答案错误地计算了 TPR。它应该是TPR=TP(TP + FN). 什么TP(TP + FP)计算是精度。