ml_grid.model_classes.H2OXGBoostClassifier

H2O XGBoost Classifier Wrapper.

This module provides a scikit-learn compatible wrapper for H2O’s XGBoostEstimator.

Classes

H2OXGBoostClassifier

Initializes the H2OXGBoostClassifier.

Module Contents

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

Bases: ml_grid.model_classes.H2OBaseClassifier.H2OBaseClassifier

Initializes the H2OXGBoostClassifier.

All keyword arguments are passed directly to the H2OXGBoostEstimator. Example args: ntrees=50, max_depth=5