Fig. 4: Performance of machine learning algorithms for classification.

a ROC curve of the classifiers. The accuracy and AUC of logistic regression were 84.8% and 0.901, respectively, which was better than those of the other classifier algorithms. b Features correlation heatmap. ROC receiver operating characteristic, AUC area under the curve, KNN k-nearest neighbor, SVM support vector machine, DT decision tree, RF random forest, GBDT gradient boosting decision tree.