/sampyl

MCMC samplers for Bayesian estimation in Python, including Metropolis-Hastings, NUTS, and Slice

Primary LanguagePythonMIT LicenseMIT

Sampyl

May 29, 2018: version 0.3

Sampyl is a package for sampling from probability distributions using MCMC methods. Similar to PyMC3 using theano to compute gradients, Sampyl uses autograd to compute gradients. However, you are free to write your own gradient functions, autograd is not necessary. This project was started as a way to use MCMC samplers by defining models purely with Python and numpy.

Sampyl includes these samplers currently:

  • Metropolis-Hastings
  • Hamiltonian
  • NUTS
  • Slice

For each sampler, you pass in a function that calculates the log probability of the distribution you wish to sample from. For the Hamiltonian and NUTS samplers, gradient log probability functions are also required. If autograd is installed, the gradients are calculated automatically. Otherwise, the samplers accept gradient log-p functions which can be used whether autograd is installed or not.

It is still under active development with more features coming soon!

Dependencies

Works for Python 2 or 3.

Currently, numpy and scipy are the only dependencies. To use the automatic gradient log-P capabilities, you will need to install autograd.

Installation

Unfortunately, there was a name collision, so use this to install from PyPI:

pip install sampyl-mcmc

Documentation

You can find the documentation at http://mcleonard.github.io/sampyl/. It is a work in progress, of course, but we'll cover all the important bits soon enough.

Tests

Tests are written for use with pytest, in the tests folder.

License

MIT