fit_earth: Fit an Earth model

Description Usage Arguments Value

Description

Fit a multivariate regression spline model using earth::earth. See earth::earth for more details.

Usage

1
2
fit_earth(st, model_name = "earth", fold_col = NULL, fold_id = NULL,
  return_model_obj = F)

Arguments

st

stacker governor. Stacking governer object with an earth model initialized. See mbgstacking::init_earth for more details.

model_name

character vector. Name/model identifier of the model to be run

fold_col

character vector. Denotes the name of the column designating the fold for crossvalidation

fold_id

Numeric. Designates the value in fold col that should be held out

return_model_obj

logical. Denotes whether the function should return the earth model object or just predictions.

Value

List object with a data.table of predictions. If return_model_obj==T, the earth command and model object are returned as well


dahcase/mbgstacking documentation built on May 20, 2019, 4:08 p.m.