Table 4.
Comparisons of nine different machine learning models.
Model | Accuracy | Average precision | AUC |
---|---|---|---|
Logistics regression | 0.909 | 0.3600 | 0.7778 |
SVC | 0.900 | 0.9769 | 0.9904 |
Decision tree | 0.992 | 0.9255 | 0.9781 |
Bagging | 0.916 | 0.9827 | 0.9933 |
GBM | 0.995 | 0.9325 | 0.9896 |
KNN | 0.895 | 0.1325 | 0.5970 |
Random tree | 0.996 | 0.9810 | 0.9923 |
XGBoost | 0.996 | 0.9788 | 0.9934 |
LightGBM | 0.996 | 0.9811 | 0.9935 |