A data scientist is training a binary classification model on an imbalanced dataset (95% negative class, 5% positive class). The model achieves 95% accuracy but only predicts the negative class for all examples. Which metric should the scientist use to evaluate model performance more appropriately?
Trap 1: F1 score
F1 score is useful but may still be high if the model predicts only the majority class; AUC-ROC is more robust.
Trap 2: Mean squared error
MSE is for regression tasks, not classification.
Trap 3: Accuracy
Accuracy can be high even if the model always predicts the majority class, which is misleading.
- A
F1 score
Why wrong: F1 score is useful but may still be high if the model predicts only the majority class; AUC-ROC is more robust.
- B
Mean squared error
Why wrong: MSE is for regression tasks, not classification.
- C
Accuracy
Why wrong: Accuracy can be high even if the model always predicts the majority class, which is misleading.
- D
AUC-ROC
AUC-ROC evaluates the model's ability to distinguish between classes regardless of threshold and is robust to imbalance.