Function reference • stackgbm

Training

stackgbm(): Model stacking for boosted trees

Inference

predict(<stackgbm>): Make predictions from a stackgbm model object

Parameter tuning

cv_param_grid(): Generate a parameter grid for cross-validation

cv_xgboost(): xgboost - parameter tuning and model selection with k-fold cross-validation and grid search

cv_lightgbm(): lightgbm - parameter tuning and model selection with k-fold cross-validation and grid search

cv_catboost(): catboost - parameter tuning and model selection with k-fold cross-validation and grid search

Wrapper functions

catboost_load_pool(): Create a dataset

catboost_train(): Train the model

catboost_predict(): Predict based on the model

lightgbm_train(): Train lightgbm model

xgboost_dmatrix(): Create xgb.DMatrix object

xgboost_train(): Train xgboost model