Skip to main content

Table 2 Comparison of the prediction accuracy of ML-based algorithms for six diseases

From: Comparison of machine learning algorithms applied to symptoms to determine infectious causes of death in children: national survey of 18,000 verbal autopsies in the Million Death Study in India

ML algorithms Pneumonia Diarrhoeal diseases Malaria Meningitis/ encephalitis Measles Fever of unknown origin
SVM 0.90 0.97 0.91 0.80 0.92 0.87
GBM 0.90 0.97 0.92 0.80 0.92 0.79
C5 0.89 0.97 0.90 0.80 0.92 0.87
ANN 0.89 0.96 0.90 0.79 0.93 0.88
kNN 0.89 0.96 0.91 0.75 0.93 0.87
CART 0.87 0.96 0.83 0.75 0.92 0.81