ml_grid.model_classes.H2ORuleFitClassifier

H2O RuleFit Classifier Wrapper.

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

Classes

H2ORuleFitClassifier

Initializes the H2ORuleFitClassifier.

Module Contents

class ml_grid.model_classes.H2ORuleFitClassifier.H2ORuleFitClassifier(**kwargs)[source]

Bases: ml_grid.model_classes.H2OBaseClassifier.H2OBaseClassifier

Initializes the H2ORuleFitClassifier.

Parameters:

**kwargs – Keyword arguments passed directly to the H2ORuleFitEstimator. Common arguments include max_rule_length=3 and model_type=’rules_and_linear’.