ml_grid.model_classes.catboost_classifier_class

CatBoost Classifier.

This module contains the CatBoostClassifierClass, which is a configuration class for the CatBoostClassifier. It provides parameter spaces for grid search and Bayesian optimization.

Classes

CatBoostClassifierClass

Initializes the CatBoostClassifierClass.

Module Contents

class ml_grid.model_classes.catboost_classifier_class.CatBoostClassifierClass(X: pandas.DataFrame | None = None, y: pandas.Series | None = None, parameter_space_size: str | None = None)

Initializes the CatBoostClassifierClass.

Parameters:

• X (Optional[pd.DataFrame]) – The input features. Defaults to None.

• y (Optional[pd.Series]) – The target variable. Defaults to None.

• parameter_space_size (Optional[str]) – The size of the parameter space. Defaults to None.

Raises:

ValueError – If parameter_space_size is not a valid key (though current implementation does not explicitly raise this).

X: pandas.DataFrame | None = None

y: pandas.Series | None = None

algorithm_implementation: catboost.CatBoostClassifier

method_name: str = 'CatBoostClassifier'

parameter_space: List[Dict[str, Any]] | Dict[str, Any]