README.md

DeepEnsemble

Implementation of 'Simple and Scalable Predictive Uncertainty Estimation using Deep Ensembles'

• Requirements

  • numpy
  • matplotlib

• Command

  • python3 main.py --epochs [epochs] --batch_size [batch_size] --fig [figure number]

Reference

1. Simple and Scalable Predictive Uncertainty Estimation using Deep Ensembles (https://arxiv.org/abs/1612.01474)
2. Dataset (https://arxiv.org/abs/1502.05336)