evalml.pipelines.components.LightGBMClassifier

class evalml.pipelines.components.LightGBMClassifier(boosting_type='gbdt', learning_rate=0.1, n_estimators=100, max_depth=0, num_leaves=31, min_child_samples=20, n_jobs=-1, random_state=0, bagging_fraction=0.9, bagging_freq=0, **kwargs)[source]

LightGBM Classifier

name = 'LightGBM Classifier'
model_family = 'lightgbm'
supported_problem_types = [<ProblemTypes.BINARY: 'binary'>, <ProblemTypes.MULTICLASS: 'multiclass'>]
hyperparameter_ranges = {'bagging_fraction': Real(low=1e-06, high=1, prior='uniform', transform='identity'), 'bagging_freq': Integer(low=0, high=1, prior='uniform', transform='identity'), 'boosting_type': ['gbdt', 'dart', 'goss', 'rf'], 'learning_rate': Real(low=1e-06, high=1, prior='uniform', transform='identity'), 'max_depth': Integer(low=0, high=10, prior='uniform', transform='identity'), 'min_child_samples': Integer(low=1, high=100, prior='uniform', transform='identity'), 'n_estimators': Integer(low=10, high=100, prior='uniform', transform='identity'), 'num_leaves': Integer(low=2, high=100, prior='uniform', transform='identity')}
default_parameters = {'bagging_fraction': 0.9, 'bagging_freq': 0, 'boosting_type': 'gbdt', 'learning_rate': 0.1, 'max_depth': 0, 'min_child_samples': 20, 'n_estimators': 100, 'n_jobs': -1, 'num_leaves': 31}

Instance attributes

SEED_MAX

SEED_MIN

feature_importance

Returns importance associated with each feature.

needs_fitting

parameters

Returns the parameters which were used to initialize the component

Methods:

__init__

Initialize self.

clone

Constructs a new component with the same parameters

describe

Describe a component and its parameters

fit

Fits component to data

load

Loads component at file path

predict

Make predictions using selected features.

predict_proba

Make probability estimates for labels.

save

Saves component at file path

Class Inheritance

Inheritance diagram of LightGBMClassifier