This folder is a direct copy of the following github:

https://github.com/y0ast/deterministic-uncertainty-quantification

It was modified for our purpose by adding and changing some features. We deleted files that were not used in our experiments to keep the folder concise and clear.

See the above repository to download NotMNIST data.

Brief overview:


1. "environment.yml" contains modules and their version to create a virtual environment where this code can be executed.

2. "two_moons.py" was used for experiments with the two moons dataset.
For example, run the following code to reproduce Figure 4

for BLNN

  python two_moons.py -md blnn -convex 2 -smooth 4  -hd 20 -ep 50 -nhl 2 -od 40 -brute 1

for DUQ

  python two_moons.py -md duq -convex 2 -ep 50

and for deep ensemble

  python two_moons_ensemble.py


3. "train_duq_fm3.py" was used for experiments with the Fashion-MNIST dataset.

For example, run the following code to reproduce Table 2

for DUQ

  python train_duq_fm3.py -md duq -od 256 -hd 150 -ep 50 -s 1 -nhl1 3

for DUQ+BLNN

  python train_duq_fm3.py -md duq -od 50 -hd 150 -smooth 3.0 -convex 0 -op GD -ep 50 -nhl1 3
