include mse plots, /close #34

src/model_modules/model_class.py

src/plotting/training_monitoring.py

src/run_modules/model_setup.py

src/run_modules/training.py

test/test_modules/test_training.py

test/test_plotting/test_training_monitoring.py