Developing a predictive model for COVID-19 in Uganda

Stuart Muwanga Sebiranda*, Peter Busch, Stephen Smith

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference proceeding contributionpeer-review

Abstract

While machine learning (ML) has been used in the fight against COVID-19, there is limited research using ML to make predictions and identify essential variables in a developing country in East Africa. With COVID-19 widening the gap between developing and developed countries, it is necessary to reduce this disparity, which is also noted as a significant UN Sustainable Development Goal. This paper aims to use ML models to compare the spread of COVID-19 in Uganda. The daily and cumulative number of COVID-19 cases are modelled using two machine learning models: linear regression and the cat boost regressor model. The cat boost model was the best-performing model with promising results. The significant variables in the cat boost model were then determined using SHAP values, which showed the number of government hospitals, GDP per capita and temperature, significantly impacted COVID-19 cases in Uganda
Original languageEnglish
Title of host publicationAustralasian Conference on Information Systems (ACIS)
Place of PublicationUniversity of Canberra
Number of pages11
Publication statusSubmitted - 19 Aug 2024
EventAustralasian Conference on Information Systems 2024
: Digital Futures for a Sustainable Society
- University of Canberra, Canberra, Australia
Duration: 4 Dec 20246 Dec 2024
Conference number: 35th
https://acis.aaisnet.org/acis2024/

Conference

ConferenceAustralasian Conference on Information Systems 2024
Abbreviated titleACIS 2024
Country/TerritoryAustralia
CityCanberra
Period4/12/246/12/24
Internet address

Keywords

  • COVID-19
  • SHAP
  • Uganda
  • Machine learning (ML)
  • Cat Boost Regressor
  • Linear Regression

Cite this