evalml.pipelines.BinaryClassificationPipeline

class evalml.pipelines.BinaryClassificationPipeline(parameters, random_state=None, random_seed=0)[source]

Pipeline subclass for all binary classification pipelines.

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

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

threshold

Threshold used to make a prediction.

Methods:

__init__

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

clone

Constructs a new pipeline with the same components, parameters, and random state.

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 BinaryClassificationPipeline