dengue-data-set

Project on predictions of dengue fever cases reported in San Juan and Iquitos (Peru) with a dataset from https://www.drivendata.org/competitions/44/dengai-predicting-disease-spread/

Code Description

The project is implemented in Python 3 and uses the following libraries/functions:

• pandas
• scikit-learn
• src.utils
• src.drop_columns
• src.ProcessData

The code performs the following steps:

1. Data Cleaning
2. Outlier Detection
3. Cyclical Encoding
4. Hyperparameter Tuning
5. Train two different models for the two cities and make predictions on the test data

The code to execute is available in main.py file.