Join GitHub today
GitHub is home to over 36 million developers working together to host and review code, manage projects, and build software together.Sign up
Add predict with Monte Carlo dropout #2132
I've added some functionality to use Monte Carlo dropout at prediction time (forum link).
This technique can be used to measure model uncertainty in neural networks.
The variance of these predictions can be used to measure how certain the model is about a prediction.
High variance = high uncertainty
Inspiration for this was gathered from two main resources:
We've been using it on a text classification problem at Max Kelsen and it has proven very valuable.
Thanks for your PR!