Learn R Programming

GeDS (version 0.2.5)

crossv_GeDS: k-fold cross-validation

Description

crossv_GeDS performs k-fold cross-validation for tuning the relevant parameters of the NGeDS, GGeDS, NGeDSgam, and NGeDSboost models.

Value

Two data frames, best_params and results. best_params contains the best combination of parameters according to the cross-validated MSE. results presents the results for each of the possible combinations of parameters, given the input parameters.

Arguments

formula

a description of the structure of the model structure, including the dependent and independent variables.

data

a data frame containing the variables referenced in the formula.

model_fun

the GeDS model to be fitted, that is, NGeDS, GGeDS, NGeDSgam or NGeDSboost.

parameters

to tune via cross-validation. These are: beta, phi and q in the case of NGeDS, GGeDS and NGeDSgam. In addition, for NGeDSboost, int.knots_init and shrinkage can also be tuned. Default values are int.knots_init_grid = c(0, 1, 2), shrinkage_grid = c(0.1, 0.5, 1), beta_grid = c(0.3, 0.5, 0.7), phi_grid = c(0.9, 0.95, 0.99), q_grid = c(2, 3)).