INNER CODE UNIT · Python

train

fabsig/GPBoost · python-package/gpboost/engine.py:52

def train(params, train_set, num_boost_round=100,
          gp_model=None, use_gp_model_for_validation=True, train_gp_model_cov_pars=True,
          valid_sets=None, valid_names=None,
          fobj=None, feval=None, init_model=None,
          feature_name='auto', categorical_feature='auto',
          early_stopping_rounds=None, evals_result=None,
          verbose_eval=True, learning_rates=None,
          keep_training_booster=False, callbacks=None):
    """Training function.

    Parameters
    ----------
    params : dict
        Parameters for training.
        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.
    num_boost_round : int, optional (default=100)

View source record →