Data dan MLPilihanL3:covid-herd-immunity-forecast
Prediksi Herd Immunity COVID-19 dengan N-BEATS
Studi dinamika kasus dan vaksinasi COVID-19 di Indonesia, dengan prediksi N-BEATS untuk kasus harian dan vaksinasi dosis lengkap.
- Peran
- Co-author dan penulis notebook dalam tim tiga orang
- Periode
- Agu 2021
Hasil
0.992
R-squared uji mundur model kasus harian
Historical forecast 20 Maret hingga 22 Agustus 2021 pada deret training, jadi pengecekan in-sample (r2_score Darts, output notebook).
0.9995
R-squared uji mundur model vaksinasi dosis lengkap
Metode historical forecast in-sample yang sama pada vaksinasi dosis lengkap kumulatif (output notebook).
208 days
Data harian yang dipakai, sejak 27 Januari 2021
Output notebook model kasus utama.
About 10 months
Menuju vaksinasi lengkap 80% dengan laju Agustus 2021
Dengan rata-rata 273.318 dosis kedua per hari (paper di repository).
Masalah
Pada Agustus 2021 Indonesia baru melewati gelombang COVID-19 terbesarnya, dan pertanyaannya adalah kapan kasus mereda dan berapa lama vaksinasi lengkap menuju herd immunity dengan laju saat itu.
Yang saya kerjakan
- 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 menguji mundur 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
Repository di GitHub
Paper, skripsi, dan model
Makalah1 Des 2021
2021 International Conference on Intelligent Cybernetics Technology & Applications (ICICyTA)
Tech stack
Python
- N-BEATS
- Darts
PyTorch
pandas
Plotly
- Matplotlib
Jupyter