Skip to main content
. 2022 May 27;17(1):33. doi: 10.5334/gh.1113

Figure 3.

Comparison of prediction performance of six machine learning classification algorithms

Comparison of prediction performance of six machine learning classification algorithms. The six algorithms include Gradient Boosting Decision Tree (GBDT), Extreme Gradient Boosting (Xg-boost), Random Forest, Light-GBM, Decision Tree, and Support Vector Machine (SVM). The ROC curve of GBDT is shown by the yellow line, the ROC curve of Xg-boost is shown by the blue line, the ROC curve of Random Forest is shown by the green line, the ROC curve of Light-GBM is shown by the orange line, the ROC curve of Decision Tree is shown by the red line, the ROC curve of SVM is shown by the black line.