Skip to content
Masked Autoregressive Flow
Branch: master
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
datasets changed the paths of where datasets are stored May 18, 2017
ml fixed likelihood calculation in conditional maf Jun 1, 2018
.gitignore ignored .DS_Store for use on macs Jul 15, 2018
LICENCE.txt added a readme and a licence May 19, 2017
README.md Updated links to arXiv and bibtex. Jun 28, 2019
collect_results.py initial commit with all the code May 17, 2017
experiments.py
pdfs.py initial commit with all the code May 17, 2017
run_experiments.py initial commit with all the code May 17, 2017
util.py initial commit with all the code May 17, 2017

README.md

Masked Autoregressive Flow for Density Estimation

Code for reproducing the experiments in the paper:

G. Papamakarios, T. Pavlakou, and I. Murray. Masked Autoregressive Flow for Density Estimation. Advances in Neural Information Processing Systems Conference. 2017. [arXiv] [bibtex]

How to run the code

To run all experiments for a particular dataset, run:

python run_experiments.py <dataset>

This will train and save all models associated with that dataset.

To evaluate all trained models and collect the results in a text file, run:

python collect_results.py <dataset>

In the above commands, <dataset> can be any of the following:

  • power
  • gas
  • hepmass
  • miniboone
  • bsds300
  • mnist
  • cifar10

You can use the commands with more than one datasets as arguments separated by a space, for example:

python run_experiments.py mnist cifar10  
python collect_results.py mnist cifar10

How to get the datasets

  1. Downdload the datasets from: https://zenodo.org/record/1161203#.Wmtf_XVl8eN
  2. Unpack the downloaded file, and place it in the same folder as the code.
  3. Make sure the code reads from the location the datasets are saved at.
  4. Run the code as described above.

All datasets used in the experiments are preprocessed versions of public datasets. None of them belongs to us. The original datasets are:

You can’t perform that action at this time.