Spatial distribution and machine learning prediction of sexually transmitted infections and associated factors among sexually active men and women in Ethiopia, evidence from EDHS 2016

Table 5 Performance of supervised machine learning algorithm models for STI prediction in Ethiopia: evidence from EDHS 2016

Confusion matrix	Naïve Bayes		Random forest		Decision tree		SVM		LR
	Observed		Observed		Observed		Observed		Observed
	STI	No_STI	STI	No_STI	STI	No_STI	STI	No_STI	STI	No_STI
Predicted
STI	120	90	138	55	122	82	123	80	110	106
No_STI	93	123	75	158	91	131	90	133	103	107
Metrics	%		%		%		%		%
Accuracy	57		69		59		60		51
95% CI	(52, 62)		(65, 74)		(55, 64)		(55, 65)		(46, 56)
Sensitivity	56		65		57		58		52
Specificity	58		74		62		62		50
Positive predictive value	57		72		60		61		51
Negative predictive value	57		68		59		60		51
AUC	58		69		56		61		54

ISSN: 1471-2334