evalml.objectives.BinaryClassificationObjective.decision_function¶
-
BinaryClassificationObjective.
decision_function
(ypred_proba, threshold=0.5, X=None)[source]¶ 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