0
$\begingroup$

I have a binary response variable (label) ๐ต in a dataset with around 50,000 observations.

The training set is somewhat imbalanced with, ๐ต๐‘–=1 making up about 33% of the observation's and ๐ต๐‘–=0 making up about 67% of the observations. Right now with XGBoost I'm getting a ROC-AUC score of around 0.67.

The response variable is binary so the baseline is 50% in term of chance, but at the same time the data is imbalanced, so if the model just guessed ๐ต๐‘–=0 it would also achieve a ROC-AUC score of 0.67. So does this indicate the model isn't doing better than chance at 0.67?

$\endgroup$

1 Answer 1

3
$\begingroup$

if the model just guessed ๐ต๐‘–=0 it would also achieve a ROC-AUC score of 0.67.

This is incorrect. The ROC curve is defined by varying a decision threshold, and so requires a probability or other confidence measure, not just a hard prediction. Guessing all points as a single class is represented in ROC space as the top-right or bottom-left corner, and doesn't give much information about the AUC.

One interpretation of AUROC is "the probability that, given a random positive instance and a random negative instance, the predicted probability (or confidence) of the positive instance is higher than that of the negative". Stated that way, it's clear that a classifier that predicts random probabilities for every instance will have 0.5 AUC, regardless of class balance.

$\endgroup$

Your Answer

By clicking โ€œPost Your Answerโ€, you agree to our terms of service and acknowledge that you have read and understand our privacy policy and code of conduct.

Not the answer you're looking for? Browse other questions tagged or ask your own question.