evalml.pipelines.XGBoostPipeline

class evalml.pipelines.XGBoostPipeline(objective, eta, min_child_weight, max_depth, impute_strategy, percent_features, number_features, n_estimators=10, n_jobs=-1, random_state=0)[source]

XGBoost Pipeline for both binary and multiclass classification

Methods

__init__

Machine learning pipeline made out of transformers and a estimator.

describe

Outputs pipeline details including component parameters

fit

Build a model

get_component

Returns component by name

predict

Make predictions using selected features.

predict_proba

Make probability estimates for labels.

score

Evaluate model performance on current and additional objectives