evalml.objectives.ROC.fit

ROC.fit(y_predicted, y_true, extra_cols=None)

Learn the objective function based on the predictions from a model.

If needs_fitting is false, this method won’t be called

Parameters

• y_predicted (list) – the predictions from the model. If needs_proba is True, it is the probability estimates

• y_true (list) – the ground truth for the predictions.

• extra_cols (pd.DataFrame) – any extra columns that are needed from training data to fit. Only provided if uses_extra_columns is True.

Returns

self