evalml.objectives.CostBenefitMatrix.decision_function

CostBenefitMatrix.decision_function(ypred_proba, threshold=0.5, X=None)

Apply a learned threshold to predicted probabilities to get predicted classes.

Parameters
  • ypred_proba (ww.DataColumn, pd.Series, np.ndarray) – The classifier’s predicted probabilities

  • threshold (float, optional) – Threshold used to make a prediction. Defaults to 0.5.

  • X (ww.DataTable, pd.DataFrame, optional) – Any extra columns that are needed from training data.

Returns

predictions