| Model | AUC of imbalanced data | AUC of balanced data |
| LR | 0.733 | 0.858 |
| SVM | 0.682 | 0.694 |
| KNN | 0.722 | 0.782 |
| DT | 0.494 | 0.646 |
| RF | 0.774 | 0.836 |
| XGboost | 0.900 | 0.914 |