myogestic.models.definitions.catboost_models.train#

myogestic.models.definitions.catboost_models.train(model, dataset, is_classifier, logger)[source]#

Train a CatBoost model.

Parameters:
  • model (_CatBoostBase) – The CatBoost model to train.

  • dataset (dict) – The dataset to train the model.

  • is_classifier (bool) – If the model is a classifier.

  • logger (CustomLogger) – The logger to use.

Returns:

The trained CatBoost model.

Return type:

_CatBoostBase