Skip to content

y_pred when metric is roc_auc #276

Answered by nnansters
mglowacki100 asked this question in Q&A
Discussion options

You must be logged in to vote

Hey @mglowacki100 , good question.

You are completely correct, roc_auc only requires the y_pred_proba values. Asking for y_pred is a consequence of us allowing multiple metrics to be calculated at once. Hence we'll need all the information those other metrics might require, such as y_pred. The way we've implemented it at this time is a bit lacking since we do expect a y_pred column to be present even when you're only estimating roc_auc. It will not be used, however, so you can pass a random values column there when only estimating roc_auc.

We are thinking about splitting up the metrics into separate runnable pieces, which would allow to only ask for parameters we actually need.

Hope that …

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by nnansters
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants