ml_grid.model_classes.H2OGAMClassifier

Attributes

logger

Classes

H2OGAMClassifier

Initializes the H2OGAMClassifier.

Module Contents

ml_grid.model_classes.H2OGAMClassifier.logger[source]
class ml_grid.model_classes.H2OGAMClassifier.H2OGAMClassifier(_suppress_low_cardinality_error=True, **kwargs)[source]

Bases: ml_grid.model_classes.H2OBaseClassifier.H2OBaseClassifier

Initializes the H2OGAMClassifier.

Parameters:

  • _suppress_low_cardinality_error (bool, optional) – If True, safely removes GAM columns with insufficient unique values to define knots. If False, raises a ValueError. Defaults to True.

  • **kwargs – Additional keyword arguments passed directly to the H2OGeneralizedAdditiveEstimator. Common arguments include family=’binomial’, gam_columns=[‘feature1’], etc.

get_params(deep: bool = True) → Dict[str, Any][source]

Gets parameters for this estimator, including the custom suppression flag.

fit(X: pandas.DataFrame, y: pandas.Series, **kwargs) → H2OGAMClassifier[source]

Fits the H2O GAM model, falling back to GLM if necessary.

shutdown()[source]

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