EnbinYang/tb_prediction_files

A multivariate multi-step LSTM forecasting model for tuberculosis incidence with model explanation

15
/ 100
Experimental

Implements a complete epidemiological forecasting pipeline with data preprocessing (kNN imputation, LASSO feature screening, random forest ranking) feeding into an ensemble architecture combining ARIMA and multi-step LSTM models. Incorporates SHAP analysis for model interpretability, enabling clinicians to understand which meteorological, economic, and social factors drive TB incidence predictions. Designed for reproducibility with raw Chinese datasets and sequential Python scripts covering imputation through final explanation.

No commits in the last 6 months.

No License Stale 6m No Package No Dependents
Maintenance 0 / 25
Adoption 7 / 25
Maturity 1 / 25
Community 7 / 25

How are scores calculated?

Stars

27

Forks

2

Language

Python

License

Last pushed

May 24, 2022

Commits (30d)

0

Get this data via API

curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/EnbinYang/tb_prediction_files"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.