r - R 中的精度、召回率和 f 度量

Question

我有一段时间没有使用 R，所以也许我只是不习惯它，但是..我在 R 中有一个表，有两个列，第一个有预测值(值可以是 0 或 1 )，第二个具有实际值(也是 0 或 1)。我需要找到召回率、精度和 f 度量，但在 R 中找不到合适的函数。(我也读过 ROCR，但我所能做的就是创建一些图，但我真的不需要图，我需要数字)。

有没有什么好的函数可以在 R 中找到精确度、召回率和 f 度量？
有什么不同的方法可以做到吗？

Answer 1

首先我创建一个数据集

> predict <- sample(c(0, 1), 20, replace=T)
> true <- sample(c(0, 1), 20, replace=T)

我想预测值中的那些 1 是检索到的。检索到的总数为

> retrieved <- sum(predict)

精度是检索到的相关实例的分数，是

> precision <- sum(predict & true) / retrieved

Recall 是检索到的相关实例的分数，是

> recall <- sum(predict & true) / sum(true)

F-measure 是 2 * 精度 * 召回率/(精度 + 召回率)是

> Fmeasure <- 2 * precision * recall / (precision + recall)