binary_classification_pipeline_mixin

Binary classification pipeline mix-in class.

Module Contents

Classes Summary

BinaryClassificationPipelineMixin

Binary classification pipeline mix-in class.

Contents

class evalml.pipelines.binary_classification_pipeline_mixin.BinaryClassificationPipelineMixin

Binary classification pipeline mix-in class.

Methods

optimize_threshold	Optimize the pipeline threshold given the objective to use. Only used for binary problems with objectives whose thresholds can be tuned.
threshold	Threshold used to make a prediction. Defaults to None.

optimize_threshold(self, X, y, y_pred_proba, objective)

Optimize the pipeline threshold given the objective to use. Only used for binary problems with objectives whose thresholds can be tuned.

Parameters

• X (pd.DataFrame) – Input features.

• y (pd.Series) – Input target values.

• y_pred_proba (pd.Series) – The predicted probabilities of the target outputted by the pipeline.

• objective (ObjectiveBase) – The objective to threshold with. Must have a tunable threshold.

Raises

ValueError – If objective is not optimizable.

property threshold(self)

Threshold used to make a prediction. Defaults to None.