This repo contains the code that predicts the total dengue cases spread
An analysis in python using sklearn and statsmodels api
Data provided by DrivenData: https://www.drivendata.org/competitions/44/dengai-predicting-disease-spread/
Notebooks:
Exploration and benchmark model [core]: this notebook includes initial data exploration, correlation plots, and decision tree benchmark model
Exploration and visuals [additional]: this notebook includes visuals from the training dataset to better understand total dengue counts over time
Time series - closer look [additional]: this notebook briefly explores ARIMA for time series forecasting and time series decomposition
Evaluating models [core]: this notebook evaluates the following algorithms on the dataset: decision tree, random forest, SVR, and linear regression
Final models [core]: this notebook showcases two final models: negative binomial regression and SVR, grid search cv is performed and models evaluated with mean absolute error and mean squared error