Table 2 The performance of different machine learning models
Machine learning | AUC | Precision |
|---|---|---|
Random forest | 0.9 | 0.696 |
Logistic regression | 0.921 | 0.754 |
SVM | 0.937 | 0.773 |
Decision tree | 0.721 | 0.323 |
GBDT | 0.946 | 0.778 |
KNN | 0.747 | 0.385 |
NN | 0.9 | 0.601 |
XGBOOST | 0.939 | 0.776 |