python - 绘制 NLTK 分类器的 ROC 曲线

Question

这本来可以回答我的问题，但它似乎是从 2011 年开始的，并且该方法已被弃用：Ntlk & Python, plotting ROC curve

在我训练过的分类器上获得 ROC 曲线的最佳方法是什么？我有一个训练有素的 nltk NaiveBayes 分类器naive_bayes，naive_bayes.classify(feature)它返回pos或neg（当然我可以得到概率prob_classify(feature)）。我有一组测试功能集，我想用它来生成我的曲线。

我听说sklearn.metrics.roc_auc_score可能会，但我无法让它工作。

如果可能的话，我真的很感激一个代码示例。

标签： pythonmachine-learningscikit-learnnltkroc