Skip to main content
. 2024 Mar 12;13:24. doi: 10.1186/s40249-024-01191-7

Table 3.

Training model fit metrics for the machine learning approaches

Model AUC Threshold Accuracy Kappa Sensitivity Specificity
LM 1 0.093 1 1 1 1
RF 1 0.093 1 1 1 1
GBM 1 0.093 1 1 1 1
DT 1 0.093 1 1 1 1
NNET 0.996 0.093 0.991 0.998 0.981 1
XGBOOST 1 0.091 1 1 1 1

AUC, area under the receiver operating characteristic curve; Threshold, optimal probability threshold for model predictions; Accuracy, overall accuracy of model predictions; Kappa, Cohen's Kappa statistic measuring prediction agreement; Sensitivity, model sensitivity in predicting presence; Specificity, model specificity in predicting absence; RF, random forest model; XGBOOST, extreme gradient boosting model; GBM, gradient boosting machine model; LM, logistic regression model; DT, decision tree model; NNET, neural network model