ml_grid.model_classes.light_gbm_class

Classes

LightGBMClassifierWrapper

Initializes the LightGBMClassifierWrapper.

Module Contents

class ml_grid.model_classes.light_gbm_class.LightGBMClassifierWrapper(X: pandas.DataFrame | None = None, y: pandas.Series | None = None, parameter_space_size: str | None = None)[source]

Initializes the LightGBMClassifierWrapper.

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[source]
y = None[source]
algorithm_implementation[source]
method_name = 'LightGBMClassifier'[source]
parameter_vector_space[source]