Skip to content
master
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
ncp
 
 
 
 
 
 
 
 

README.md

Noise Contrastive Priors

This project contains the source code for Noise Contrastive Priors (NCP) by Danijar Hafner, Dustin Tran, Alex Irpan, Timothy Lillicrap, James Davidson.

Running the code

Install dependencies:

pip3 install numpy tensorflow tensorflow_probability matplotlib ruamel.yaml

Active learning on the toy data set:

python3 -m ncp.scripts.toy_active --seeds 20 --logdir /path/to/logdir

Active learning on the flights delay data set:

python3 -m ncp.scripts.flights_active \
  --seeds 10 \
  --dataset /path/to/dataset \
  --logdir /path/to/logdir

The data set path should be a file prefix for four Numpy files named <dataset>-train-inputs.npy, <dataset>-train-targets.npy, <dataset>-test-inputs.npy, <dataset>-test-targets.npy.

About

Reliable Uncertainty Estimates in Deep Neural Networks using Noise Contrastive Priors

Topics

Resources

License

Releases

No releases published

Packages

No packages published

Languages

You can’t perform that action at this time.