Skip to content

Lejeune-Lab/model-calibration

 
 

Repository files navigation

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

About

The code used to generate the results in the manuscript:

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 100.0%