R/model_info.R

In baguette: Efficient Model Functions for Bagging

baguette_models <- c("CART", "C5.0", "MARS", "nnet")

# We want to default some arguments for different models
model_defaults <-
  list(
    CART = list(cp = 0, xval = 0, minsplit = 20, maxdepth = 30, model = FALSE),
    "model rules" = list(),
    "C5.0" = list(minCases = 2),
    MARS = list(pmethod = "none", nprune = NULL, degree = 1),
    nnet = list(decay = 0, size = 10, maxit = 1000, MaxNWts = 10^5)
  )

# Enumerate the possible arguments in the fit or control functions that can
# be modified by the user. This could be done programatically to protect against
# changes but each of the underlying packages is pretty mature and there is a
# small likelihood of them changing.

model_args <-
  list(
    CART = c('method', 'parms', 'cost',
             # control function arguments:
             'minsplit', 'minbucket', 'cp',
             'maxcompete', 'maxsurrogate', 'usesurrogate', 'xval',
             'surrogatestyle', 'maxdepth'),
    "model rules" = c('unbiased', 'rules', 'extrapolation', 'sample'),
    "C5.0" = c('rules', 'costs', 'subset', 'bands', 'winnow', 'noGlobalPruning',
               'CF', 'minCases', 'fuzzyThreshold', 'sample'),
    MARS = c('pmethod', 'trace', 'glm', 'degree', 'nprune', 'nfold', 'ncross',
             'stratify', 'varmod.method', 'varmod.exponent', 'varmod.conv',
             'varmod.clamp', 'varmod.minspan', 'Scale.y'),
    nnet = c("linout", "entropy", "softmax", "censored", "skip", "rang", "maxit")
  )