ml_grid.model_classes.H2OGBMClassifier

Attributes

PARAM_SPACE_GRID

PARAM_SPACE_BAYES

Classes

H2OGBMClassifier

Initializes the H2OGBMClassifier.

Module Contents

ml_grid.model_classes.H2OGBMClassifier.PARAM_SPACE_GRID[source]
ml_grid.model_classes.H2OGBMClassifier.PARAM_SPACE_BAYES[source]
class ml_grid.model_classes.H2OGBMClassifier.H2OGBMClassifier(parameter_space_size='small', **kwargs)[source]

Bases: ml_grid.model_classes.H2OBaseClassifier.H2OBaseClassifier

Initializes the H2OGBMClassifier.

All keyword arguments are passed directly to the H2OGradientBoostingEstimator. Example args: ntrees=50, max_depth=5, learn_rate=0.1, seed=1

parameter_space_size = 'small'[source]