I am trying to put IFERROR condition in R like Excel IFERROR Function. I am building a random forest model. To fine tune, i use tuneRF function. It helps to give optimal mtry parameter.
#Selecting Optimal MTRY parameter
mtry <- tuneRF(dat3[, -36], dat3[,36], ntreeTry=1000, stepFactor=1.5,improve=0.01, trace=TRUE, plot=TRUE)
best.m <- mtry[mtry[, 2] == min(mtry[, 2]), 1]
SOMETIMES, the above function returns error if OOB error would not improve in different iterations.
Error in if (Improve > improve) { : missing value where TRUE/FALSE needed.
Next Step : If the above function works fine, i use the value of best.m in the code below.
No ERROR in tuneRF function - Run the code below.
rf <-randomForest(classe~.,data=dat3, mtry=best.m, importance=TRUE,ntree=1000)
ERROR in tuneRF function - Run the code below.
#Train Random Forest
rf <-randomForest(classe~.,data=dat3, importance=TRUE,ntree=1000)
Thanks in anticipation! Any help would be highly appreciated.
You need to use try
or tryCatch
. This should work:
mtry <- try(tuneRF(dat3[, -36], dat3[,36], ntreeTry=1000,
stepFactor=1.5,improve=0.01, trace=TRUE, plot=TRUE))
if (!inherits(mtry, "try-error")) {
best.m <- mtry[mtry[, 2] == min(mtry[, 2]), 1]
rf <- randomForest(classe~.,data=dat3, mtry=best.m, importance=TRUE,ntree=1000)
} else {
rf <- randomForest(classe~.,data=dat3, importance=TRUE,ntree=1000)
}
However, the error given may represent a bug in the tuneRF
function. Can you give a reproducible example, i.e. with a minimal dataset that will produce the error?
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With