ml_grid.model_classes.xgb_classifier_class

Classes

XGB_class_class

Initializes the XGB_class_class.

Module Contents

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

Initializes the XGB_class_class.

The XGB_class_class wraps the XGBoost classifier algorithm, allowing for easy configuration and use within a grid search or Bayesian optimization framework by setting up a customizable parameter space.

Parameters:

• X (Optional[pd.DataFrame]) – Feature matrix for training. Defaults to None.

• y (Optional[pd.Series]) – Target vector for training. Defaults to None.

• parameter_space_size (Optional[str]) – Size of the parameter space for optimization. Defaults to None.

X = None

y = None

algorithm_implementation

method_name = 'XGBClassifier'

parameter_vector_space