evalml.pipelines.ModeBaselineBinaryPipeline.optimize_threshold¶
-
ModeBaselineBinaryPipeline.optimize_threshold(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 (ww.DataTable) – Input features
y (ww.DataColumn) – Input target values
y_pred_proba (ww.DataColumn) – The predicted probabilities of the target outputted by the pipeline
objective (ObjectiveBase) – The objective to threshold with. Must have a tunable threshold.