Model | Performance metrics of six machine learning algorithms in the training set | ||||
---|---|---|---|---|---|
AUC | Accuracy | F1 | Precision | Recall | |
Decisoin tree | 0.691 | 0.693 | 0.695 | 0.698 | 0.693 |
Gradient boosting | 0.676 | 0.657 | 0.659 | 0.662 | 0.657 |
Logistic regression | 0.590 | 0.633 | 0.559 | 0.598 | 0.633 |
Neural network | 0.711 | 0.699 | 0.682 | 0.690 | 0.699 |
Random forest | 0.835 | 0.774 | 0.767 | 0.771 | 0.774 |
Support vector machine | 0.582 | 0.593 | 0.589 | 0.586 | 0.593 |