Model calibration

This folder contains two Jupyter python notebooks demonstrating StellarGraph model calibration for binary (calibration-link-prediction.ipynb) and multi-class classification (calibration-node-classification.ipynb) problems.

References

1. On Calibration of Modern Neural Networks. C. Guo, G. Pleiss, Y. Sun, and K. Q. Weinberger. ICML (link)

Table of contents

• Calibrating a GraphSAGE link prediction model
• Calibrating a GraphSAGE node classification model