Skip to content

Toy examples to evaluate different uncertainty estimation methods for neural networks.

Notifications You must be signed in to change notification settings

mateoguaman/uncertainty_playground

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 

Repository files navigation

Toy examples to evaluate different uncertainty estimation methods for neural networks.

To get started, go inside mlp/create_data.py, change the directory where you'd like to store your toy dataset at the bottom of the script, and run this script. Then, there are three current implementations: 1. An MLP predictor that does not estimate variance (mlp/driver.py) 2. An MLP predictor that predicts the mean of a Gaussian and a variance (mlp/gaussian_driver.py) 3. A predictor that uses an ensemble of MLPs to predict single values, from which we obtain a std deviation of the predictions to quantify uncertainty (mlp/ensemble_driver.py)

To run any of these files, first go into the scripts and change the directories for the dataset at the bottom of the script. Also, change the value of USE_WANDB depending on your preference. Leaving it to true will require WANDB to be first setup, which you can do by following instructions on their website.

About

Toy examples to evaluate different uncertainty estimation methods for neural networks.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published