ml_grid.model_classes.H2ODeepLearningClassifier

Attributes

PARAM_SPACE_GRID

PARAM_SPACE_BAYES

Classes

H2ODeepLearningClassifier

Initializes the H2ODeepLearningClassifier.

Module Contents

ml_grid.model_classes.H2ODeepLearningClassifier.PARAM_SPACE_GRID[source]
ml_grid.model_classes.H2ODeepLearningClassifier.PARAM_SPACE_BAYES[source]
class ml_grid.model_classes.H2ODeepLearningClassifier.H2ODeepLearningClassifier(hidden=None, hidden_config=None, parameter_space_size='small', **kwargs)[source]

Bases: ml_grid.model_classes.H2OBaseClassifier.H2OBaseClassifier

Initializes the H2ODeepLearningClassifier.

It allows specifying hidden layers either directly via ‘hidden’ or through a predefined configuration name ‘hidden_config’.

Parameters:

  • hidden (list, optional) – A list of integers specifying the number of neurons for each hidden layer. Defaults to None.

  • hidden_config (str, optional) – A string key (‘small’, ‘medium’, ‘large’) to select a predefined hidden layer architecture. Defaults to None.

  • **kwargs – Additional keyword arguments passed to the H2ODeepLearningEstimator.

hidden = None[source]
hidden_config = None[source]
parameter_space_size = 'small'[source]