Table 4.
Performance metrics of six machine learning algorithms in the training set | |||||
---|---|---|---|---|---|
Model | AUC | Acc | F1 | Precision | Recall |
Gradient Boosting | 0.810 | 0.659 | 0.533 | 0.533 | 0.533 |
Random Forest | 0.791 | 0.720 | 0.566 | 0.652 | 0.500 |
Decision Tree | 0.665 | 0.683 | 0.536 | 0.577 | 0.500 |
Neural Network | 0.643 | 0.622 | 0.415 | 0.478 | 0.367 |
Support Vector Machine | 0.547 | 0.622 | 0.205 | 0.444 | 0.133 |
Logistic Regression | 0.506 | 0.598 | 0.195 | 0.364 | 0.133 |
AUC: area under curve; Acc: Accuracy.