Table 1.
Recall and F1-Score of the Models in the Training Set and Test Set
| Model | Training set |
Test set |
||
|---|---|---|---|---|
| Recall | F1-score | Recall | F1-score | |
| SVM | 0.9648 | 0.9516 | 0.9641 | 0.9506 |
| LR | 0.9603 | 0.9490 | 0.9599 | 0.9488 |
| NB | 0.9480 | 0.9426 | 0.9524 | 0.9425 |
| DT | 0.9743 | 0.9567 | 0.9616 | 0.9469 |
| RF | 0.9656 | 0.9533 | 0.9641 | 0.9517 |
| GBDT | 0.9767 | 0.9672 | 0.9666 | 0.9574 |
| Adaboost | 0.9782 | 0.9674 | 0.9674 | 0.9554 |
| XGBoost | 0.9715 | 0.9560 | 0.9699 | 0.9582 |
Adaboost, adaptive boosting; DT, decision tree; GBDT, gradient boosting decision tree; LR, logistic regression; NB, naive Bayes; RF, random forest; SVM, support vector machine; XGBoost, eXtreme Gradient Boosting.