Skip to content

jzjosuerivera71/ih_datamadpt0420_final_project

Repository files navigation

LSTM neural network pipeline for evolution forecast of COVID-19.

Image

Introduction.

Problem description:

The main idea of the project is to train several recurrent neural networks to be able to predict the evolution of Covid19 as accurately as possible. This will help us to make better decisions about population control. Training and testing dataset is from USA.

Original Model.

From this differntial equation:

we can deduce the solution that we use to do the sigmoidal regression. This regression is the most used for this type of analisis.

This model is very useful but there is some limitations, especialy in the fluctuations of the growth.

##Metodology.

LSTM neural Network Model

This option is better to predict the evolution of the disease because the prediction is adjusted according to the previous data. Additionally this model is easy to train.

Image

Pipeline dependencies:

  • numpy
  • pandas
  • pytorch
  • matplotlib
  • datetime
  • scipy
  • sklearn
  • streamlit

Pipeline componets:

  • 3 modules (preprocesing, analisis, models)
  • 1 LSTM class
  • 2 train neural networks
  • 3 analisis functions
  • 4 preprocesing functions

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages