COVID-19 Case Prediction Model Comparison
COVID-19 Case Prediction Model Comparison
This page displays a 7-day comparison of COVID-19 case predictions using four different models: LSTM/GRU (Long Short-Term Memory/Gated Recurrent Unit), ARIMA (AutoRegressive Integrated Moving Average), Random Forest, and XGBoost. The graphs below show our predictions against the actual reported cases and compare the performance of all models.
Model Performance
The performance of each model is measured using the Mean Absolute Percentage Error (MAPE). Lower values indicate better performance.
LSTM/GRU
ARIMA
Random Forest
XGBoost
7-Day Model Comparison
This chart displays the actual cases and predicted number of COVID-19 cases for the last 7 days, comparing all four models: LSTM/GRU, ARIMA, Random Forest, and XGBoost.
Methodology
We used four different models for time series forecasting:
- LSTM/GRU (Long Short-Term Memory/Gated Recurrent Unit) neural network model
- ARIMA (AutoRegressive Integrated Moving Average) model
- Random Forest Regressor
- XGBoost Regressor
All models were trained on COVID-19 case data from the WHO dataset. Our prediction pipeline followed these steps:
- Daily data collection from the WHO COVID-19 dataset
- Data preprocessing and cleaning
- Feature creation with a sequence length of 90 days
- Model training with the latest data
- 7-day hindcast generation for all models
- Comparison of predictions with actual reported cases
- Calculation of Mean Absolute Percentage Error (MAPE) for each model
For more details on our methodology, please visit our GitHub repository.