Application of PINN to SIR modeling

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

(2025/03/09) The SEIR model, a common extension of the SIR model, has been added. Thanks to Abhishek Soni!

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.