ml_grid.model_classes.H2OAutoMLClassifier

Attributes

logger

Classes

H2OAutoMLClassifier

Initializes the H2OAutoMLClassifier.

Module Contents

ml_grid.model_classes.H2OAutoMLClassifier.logger

class ml_grid.model_classes.H2OAutoMLClassifier.H2OAutoMLClassifier(**kwargs)

Bases: ml_grid.model_classes.H2OBaseClassifier.H2OBaseClassifier

Initializes the H2OAutoMLClassifier.

Note: H2OAutoML is not a standard estimator, so we use a placeholder in the base class and manage the AutoML process within the fit method.

automl = None

fit(X: pandas.DataFrame, y: pandas.Series, **kwargs) → H2OAutoMLClassifier

Fits the H2O AutoML process.

If the dataset is too small or AutoML fails to find a leader model, it gracefully falls back to a simple GLM model.

Parameters:

• X (pd.DataFrame) – The feature matrix.

• y (pd.Series) – The target vector.

• **kwargs – Additional keyword arguments (not used by this implementation).

Returns:

The fitted classifier instance.

Return type:

H2OAutoMLClassifier

shutdown()

Shuts down the H2O cluster using the base class’s safe logic.