Application of PINN to SIR modeling

This is an example of application of physics informed neural networks (PINNs). If you want to know more details, please refer to Final Report.pdf in this repository that contains backgrounds, methods, results, and discussion.

Dependency

• torch==1.13.1

How to use

Just run the main.py!

Results

This means that my model was well optimized.

My PINN solves the system of ODEs and plots a figure like that.

ODEsolver is the most useful tool of solving a system of ODEs. The tool plots a figure like that.