evalml.pipelines.ClassificationPipeline

class evalml.pipelines.ClassificationPipeline(parameters, random_state=0)[source]

Pipeline subclass for all classification pipelines.

name = 'Classification Pipeline'
custom_name = None
problem_type = None

Instance attributes

classes_

Gets the class names for the problem.

feature_importance

Return importance associated with each feature.

parameters

Returns parameter dictionary for this pipeline

Methods:

__init__

Machine learning classification pipeline made out of transformers and a classifier.

clone

Constructs a new pipeline with the same parameters and components.

compute_estimator_features

Transforms the data by applying all pre-processing components.

describe

Outputs pipeline details including component parameters

fit

Build a classification model. For string and categorical targets, classes are sorted

get_component

Returns component by name

graph

Generate an image representing the pipeline graph

graph_feature_importance

Generate a bar graph of the pipeline’s feature importance

load

Loads pipeline at file path

predict

Make predictions using selected features.

predict_proba

Make probability estimates for labels.

save

Saves pipeline at file path

score

Evaluate model performance on objectives

Class Inheritance

Inheritance diagram of ClassificationPipeline