R/trainRF.R

In sgvignali/SDMtune: Species Distribution Model Selection

trainRF <- function(data,
                    mtry = NULL,
                    ntree = 500,
                    nodesize = 1) {

  result <- SDMmodel(data = data)

  if (is.null(mtry))
    mtry <- floor(sqrt(ncol(data@data)))

  x <- data@data
  p <- data@pa
  model <- randomForest::randomForest(x = x,
                                      y = as.factor(p),
                                      mtry = mtry,
                                      ntree = ntree)

  model_object <- RF(mtry = model$mtry,
                     ntree = model$ntree,
                     nodesize = nodesize,
                     model = model)

  result@model <- model_object

  result
}