Jurnal INFOTEL
Vol 16 No 4 (2024): November 2024

Application of Ensemble Machine Learning for Infectious Diseases with Vaccine Intervention: A Global COVID-19 Case Study

Egi Safitri (Institut Informatika dan Bisnis Darmajaya, Indonesia)
Ruki Rizalnul Fikri (Institut Informatika dan Bisnis Darmajaya, Indonesia)
Rini Nurlistiani (Institut Informatika dan Bisnis Darmajaya, Indonesia)



Article Info

Publish Date
23 Dec 2024

Abstract

The COVID-19 pandemic has posed significant challenges worldwide, especially in controlling the spread of the disease through vaccination and active case monitoring. This study aims to evaluate the effectiveness of various ensemble machine-learning models in predicting the number of daily vaccinations and the number of active cases of COVID-19 based on global data. The models used include Random Forest, Bagging, Gradient Boosting Machine (GBM), AdaBoost, and XGBoost. The evaluation results show that Random Forest provides the best performance in predicting both the number of daily vaccinations and active COVID-19 cases, with a MSE value of 4.7e+09, MAE of 16,971.1, and RMSE of 68,557.2 for daily vaccinations, as well as an R² Score of 0.989, indicating a high ability to explain data variability. The Bagging model also showed excellent results with MSE of 4.78e+09 and MAE of 17,039.8. In contrast, the AdaBoost model performed the worst in predicting both variables, with an MSE of 5.54e+10 and an MAE of 106,228.6. These findings suggest that Random Forest and Bagging are superior models for predicting the number of daily vaccinations and active COVID-19 cases. This study provides important insights into using machine learning to predict vaccination effectiveness and active case dynamics, aiding decision-making in global pandemic control efforts.

Copyrights © 2024






Journal Info

Abbrev

infotel

Publisher

Subject

Computer Science & IT Electrical & Electronics Engineering

Description

Jurnal INFOTEL is a scientific journal published by Lembaga Penelitian dan Pengabdian Masyarakat (LPPM) of Institut Teknologi Telkom Purwokerto, Indonesia. Jurnal INFOTEL covers the field of informatics, telecommunication, and electronics. First published in 2009 for a printed version and published ...