r - Quanteda 文本模型：是否可以为 svm 和朴素贝叶斯计算 AUC 并绘制 ROC 曲线？

Question

我正在使用包中svm的Naive Bayes算法quanteda text models。是否可以计算AUC和绘图ROC curves？如果您建议任何示例教程，我将不胜感激。

标签： rsvmnaivebayesquanteda

Answer 1

我用过这样的东西，它奏效了。我很感谢 Ken Benoit 在评论中的回应。

nb_quanteda.prob <- predict(tmod_nb,type="probability",  newdata = dfmat_unknown_matched)  
pred_nb_qaunteda <- prediction(as.numeric(nb_quanteda.prob[,-1]), Truth)  
perf_nb_qaunteda <- performance(pred_nb_qaunteda,"tpr","fpr")  
plot( perf_nb_qaunteda, col="darkblue", main="Roc nb")