R/mtryOptimization.R

In etiennelalechere/maturitymodelling: Forest maturity modelling

#' @title mtry optimization
#' @description Function to define mtry random forest parameter.
#' @usage mtryOptimization(response,predictors,nrep)
#' @param response predicted variable.
#' @param predictors set of predictors.
#' @param nrep number of iteration.
#' @return Plot the mtry value that minimize the mean out-of-bag error rate.
#' @export
#' @import randomForest
#' @examples # mtryOptimization(response,predictors,100)
mtryOptimization<-function(response,predictors,nrep,...){

  rf_tuned=tuneRF(predictors, response, ntreeTry=1000, improve=0.05, plot=F, doBest=FALSE)
  for(i in 1:nrep){
    print(i)
    rf_tuned=cbind(rf_tuned,tuneRF(predictors, response, ntreeTry=1000, improve=0.05, plot=F, doBest=FALSE))
  }
  rf_tuned
  tun=apply(rf_tuned[,-1],FUN=mean,MARGIN=1)
  plot(tun,ylab="OOBError",xlab="mtry")

  return(tun)

}