Session D: Automatic Control, IT and Artificial Intelligence

ML algorithm for COVID-19 outcome prediction based on demographic features in North Macedonia

Maja Kukuseva Paneva¹*, Natasha Stojkovikj¹, Vasko Kokalanov¹, Vladimir Milićević²
1 Goce Delcev University, Faculty of Computer Science, Stip, R. N. Macedonia
2 University of Kragujevac, Faculty of Mechanical and Civil Engineering in Kraljevo, Serbia
Corresponding author: Maja Kukuseva Paneva · maja.kukuseva@ugd.edu.mk

Abstract

This study presents a machine learning approach for modeling and predicting COVID-19 patient outcomes in the Republic North Macedonia based on demographic characteristics. A decision tree-based classification model has been developed using algorithms implemented in WEKA environment. The model constructs the decision tree by selecting random subsets of features at each node and determines the optimal split points based on information gain. The results indicate that age is the most significant predictor of patient outcome, serving as the root node and primary splitting attributes in the decision tree. Individuals below 60 years of age are classified as recovered, while higher age groups exhibit an increased probability of death. The gender attribute appears as a secondary factor, contributing to more refined predictions within specific age intervals. In addition to its predictive capability, the model offers a high level of interpretability, enabling clear visualization of decision rules and facilitating understanding of the relationship between demographic factors and disease outcomes. These findings highlight the importance of age- based risk stratification and demonstrate the potential of machine learning models as decision support tools in public health management.

Keywords

Machine learningCOVID-19Decision tree

Cite this paper

Recommended citation · Engineering TODAY style
M. K. Paneva, N. Stojkovikj, V. Kokalanov, and V. Milićević, “ML algorithm for COVID-19 outcome prediction based on demographic features in North Macedonia”, Proceedings of the XII International Triennial Conference Engineering TODAY (ET 2026), Vrnjačka Banja (Serbia), pp. D81–D85, https://doi.org/10.46793/ET26.D10KP, (2026)