/model-calibration

The code used to generate the results in the manuscript:

Primary LanguageJupyter NotebookMIT LicenseMIT

model-calibration

This repo contain the code and data used to generate the results in our paper Investigating Deep Learning Model Calibration for Classification Problems in Mechanics

  • To retrain ML models used in the manuscript run model-training.ipynb
  • To apply Temperature Scaling on Neural Network predictions run temp-scaling.ipynb
  • To generate figures 3-5 run figure-generation.ipynb

For further information please feel free to contact the authors:

Saeed Mohammadzadeh: saeedmhz@bu.edu

Peerasait Prachaseree: pprachas@bu.edu

Emma Lejeune: elejeune@bu.edu