INNER CODE UNIT · Python
cv
fabsig/GPBoost · python-package/gpboost/engine.py:584
def cv(params, train_set, gp_model=None, num_boost_round=1000, early_stopping_rounds=None,
folds=None, nfold=5, metric=None, verbose_eval=None,
use_gp_model_for_validation=True, fit_GP_cov_pars_OOS=False, train_gp_model_cov_pars=True,
stratified=False, shuffle=True, fobj=None, feval=None, init_model=None,
feature_name='auto', categorical_feature='auto', fpreproc=None, show_stdv=False, seed=0,
callbacks=None, eval_train_metric=False,
return_cvbooster=False, metrics=None):
"""Perform cross-validation for choosing number of boosting iterations.
Parameters
----------
params : dict
Parameters for Booster.
See https://github.com/fabsig/GPBoost/blob/master/docs/Main_parameters.rst#tuning-parameters--hyperparameters-for-the-tree-boosting-part
train_set : Dataset
Data to be trained on.
gp_model : GPModel or None, optional (default=None)
GPModel object for the GPBoost algorithm