How to calculate the AUC value for a ranger RF model ?
Hi Biostars community,
How can I calculate the AUC value for a ranger model ? Ranger is a fast implementation of randomForest algorithm in R. I'm using the following code to build the ranger model for classification purposes, and get the prediction from the model:
#Build the model using ranger() function
ranger.model <- ranger(formula, data = data_train, importance = 'impurity', write.forest = TRUE, num.trees = 3000, mtry = sqrt(length(currentComb)), classification = TRUE)
#get the prediction for the ranger model
pred.data <- predict(ranger.model, dat = data_test,)
table(pred.data$predictions)
But I dont know how to calculate the AUC value
Any idea ?
• 5,741 views
•
link
2 answers
Hi debitboro, By any chance, did you get this answer?
• 0 views
•
link
Hi debitboro, you can try the following code.
rocRFall=roc(as.factor(pre.label),pred.data)
aucRF=pROC::auc(rocRFall)
print(aucRF)
• 0 views
•
link
Log in to answer this question.