本文介绍了在R中计算精度,召回率和F1分数的简便方法的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
我在R中使用 rpart
分类器。问题是-我想在测试数据上测试经过训练的分类器。很好-我可以使用 predict.rpart
函数。
I am using an rpart
classifier in R. The question is - I would want to test the trained classifier on a test data. This is fine - I can use the predict.rpart
function.
但是我也想计算精度,回想一下和F1分数。
But I also want to calculate precision, recall and F1 score.
我的问题是-我是否必须为自己编写函数,或者R或CRAN库中是否有为此编写的函数?
My question is - do I have to write functions for those myself, or is there any function in R or any of CRAN libraries for that?
推荐答案
库将计算所有这些以及更多内容(另请参见):
The ROCR library calculates all these and more (see also http://rocr.bioinf.mpi-sb.mpg.de):
library (ROCR);
...
y <- ... # logical array of positive / negative cases
predictions <- ... # array of predictions
pred <- prediction(predictions, y);
# Recall-Precision curve
RP.perf <- performance(pred, "prec", "rec");
plot (RP.perf);
# ROC curve
ROC.perf <- performance(pred, "tpr", "fpr");
plot (ROC.perf);
# ROC area under the curve
auc.tmp <- performance(pred,"auc");
auc <- as.numeric([email protected])
...
这篇关于在R中计算精度,召回率和F1分数的简便方法的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持!