evalml.pipelines.BaselineBinaryPipeline.optimize_threshold

BaselineBinaryPipeline.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.