Prediksi Herd Immunity COVID-19 dengan N-BEATS
Studi historis N-BEATS tentang kasus dan vaksinasi COVID-19 Indonesia pada 2021. Nilai R-squared berasal dari pengecekan in-sample, bukan data uji terpisah.
Hasil utama
0.992
R-squared model kasus harian, historical forecast in-sample
0.9995
R-squared model vaksinasi dosis lengkap, historical forecast in-sample
208 days
Data harian yang dipakai, sejak 27 Januari 2021
- Co-author dan penulis notebook dalam tim tiga orang
- Agu 2021
- Pilihan
- L3:covid-herd-immunity-forecast
Hasil
0.992
R-squared model kasus harian, historical forecast in-sample
0.9995
R-squared model vaksinasi dosis lengkap, historical forecast in-sample
208 days
Data harian yang dipakai, sejak 27 Januari 2021
About 10 months
Menuju vaksinasi lengkap 80% dengan laju Agustus 2021
Masalah
Pada Agustus 2021, Indonesia baru melewati gelombang COVID-19 terbesarnya. Pertanyaan yang dikaji adalah kapan kasus akan mereda dan berapa lama waktu yang dibutuhkan untuk mencapai vaksinasi lengkap menuju herd immunity dengan laju saat itu.
Kontribusi
- Saya menggabungkan data harian kasus, kesembuhan, dan vaksinasi sejak Januari 2021 dengan data populasi, data mobilitas, serta bobot kebijakan dan hari libur.
- Saya melatih model N-BEATS di Darts untuk kasus harian (dengan mobilitas sebagai kovariat) dan vaksinasi dosis lengkap, lalu melakukan backtest dengan historical forecast sejak Maret 2021.
- Tim mengolah prediksi tersebut menjadi estimasi kapan kasus mencapai titik terendah dan berapa lama vaksinasi lengkap 80% tercapai, yang dipublikasikan di IEEE ICICyTA 2021.
Galeri
Kode dan riset
Repositori di GitHub
Makalah, skripsi, dan model
2021 International Conference on Intelligent Cybernetics Technology & Applications (ICICyTA)
Tech stack
Python
PyTorch
pandas
Plotly
Matplotlib
Jupyter
- N-BEATS
- Darts