Skip to contents

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

Utility functions

is_installed_xgboost()
Is xgboost installed?
is_installed_lightgbm()
Is lightgbm installed?
is_installed_catboost()
Is catboost installed?