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