Code for the project with the SIR model in TMA4183 Optimization 2 Spring 2021 at NTNU

Relevant article avalibe from: https://www.sciencedirect.com/science/article/abs/pii/S0303264708001044

Project was done with Ulrik R.F. and Tor Ola O.S.

Files:

Opt_2___Optimal_vaccination.pdf : the project report.
SIR_Optimal_Control_classes.py : Script containg the classes used to build up and solve a controlproblem in our case.
SIR_optimal_control_gradient_method.py : Script implementing a projected gradient method for our control problem.
plotingfunctions.py : Script containing the functions to plot.
main.py : The main script to control every thing from for the optimization part.

To use optimization part run the main.py script.
To simulate different vacciation protocoles/strategies use the SIRsolver.m script.