evalml.model_understanding.precision_recall_curve

evalml.model_understanding.precision_recall_curve(y_true, y_pred_proba)[source]

Given labels and binary classifier predicted probabilities, compute and return the data representing a precision-recall curve.

Parameters
  • y_true (ww.DataColumn, pd.Series or np.ndarray) – True binary labels.

  • y_pred_proba (ww.DataColumn, pd.Series or np.ndarray) – Predictions from a binary classifier, before thresholding has been applied. Note this should be the predicted probability for the “true” label.

Returns

Dictionary containing metrics used to generate a precision-recall plot, with the following keys:

  • precision: Precision values.

  • recall: Recall values.

  • thresholds: Threshold values used to produce the precision and recall.

  • auc_score: The area under the ROC curve.

Return type

list