evalml.objectives.AUC.decision_function

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

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

Parameters

  • ypred_proba (list) – The classifier’s predicted probabilities

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

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

Returns

predictions

evalml.objectives.AUC.calculate_percent_difference evalml.objectives.AUC.objective_function