plot.roc pour multiclass.roc dans le paquet pROC?

Question

J'essaie de tracer des courbes ROC multiclasses mais je n'ai rien trouvé de fructueux dans le paquet pROC. Voici un code de début:

data(iris) 
library(randomForest) 
library(pROC) 
set.seed(1000) 
# 3-class in response variable 
rf = randomForest(Species~., data = iris, ntree = 100) 
# predict(.., type = 'prob') returns a probability matrix 
predictions <- as.numeric(predict(rf, iris, type = 'response')) 
roc.multi <- multiclass.roc(iris$Species, predictions) 
auc(roc.multi) 

Comment tracer les courbes ROC pour des classes individuelles?

Answer 1

Vérifiez les noms de roc.multi, vous devriez trouver un nom appelé rocs, qui stocke les informations de courbe roc individuels pour chaque classe.

, vous pouvez utiliser plot.roc et lines.roc pour visualiser tous:

rs <- roc.multi[['rocs']] 
plot.roc(rs[[1]]) 
sapply(2:length(rs),function(i) lines.roc(rs[[i]],col=i))