Извлечь показатели производительности классификатора из объекта прогнозирования ROCR

Question

Я использую ROCR для получения измерений производительности различных алгоритмов классификации в моем наборе данных. В то время как я могу извлечь ППК метрику относительно легко, используя следующий код:

predictions <- predict(rfmodel, test, type="prob") 
pred.obj <- prediction(predictions[,2], data$response) 
rfperf <- performance(pred.obj, "tpr","fpr") 
print(sprintf("random forest AUC %f", as.numeric(performance(pred.obj,"auc")@y.values))) 

У меня есть проблемы с извлечением другого извлечения других мер, таких как точность, напомним, F и т.д., как единое значение, которое я мог бы включить в таблицу в моем бумага. Я пробовал следующее:

> p <- performance(pred.obj,"prec", "rec") 
> as.numeric(p@x.values) 
Error: (list) object cannot be coerced to type 'double' 

Я могу построить значения, но это не то, что я хотел бы. Есть идеи?

Answer 1

Проблема заключается в выходе списка. Попробуйте as.numeric (unlist (p@x.values))

Answer 2

Это связано со структурой возвращаемого объекта функции perf. Использование str на нем поможет.

p@x.values ​​- это список, содержащий один элемент, который является вектором числовых значений. Просто используйте

p@x.values[[1]] 

для извлечения вектора.