Table 1 Comparison of machine learning models.
Model | AUC | CA | F1 | Precision | LogLoss | Specificity |
|---|---|---|---|---|---|---|
Random forest | 0.936 | 0.849 | 0.836 | 0.853 | 0.400 | 0.659 |
Naive Bayes | 0.630 | 0.661 | 0.659 | 0.657 | 0.622 | 0.488 |
Neural network | 0.556 | 0.720 | 0.602 | 0.518 | 0.593 | 0.280 |
Gradient boosting | 0.950 | 0.862 | 0.851 | 0.868 | 0.368 | 0.682 |