Skip to main content

Efficiently sample from the Polya-Gamma distribution using NumPy's Generator interface

Project description

polya-gamma

Efficiently sample from the Polya-Gamma distribution using NumPy's Generator interface.

Dependencies

  • Numpy >= 1.17

Installation

$ pip install polyagamma

Example

polyagamma can act as a drop-in replacement for numpy's Generator class.

import numpy as np

from polyagamma import default_rng, Generator

g = Generator(np.random.PCG64())  # or use default_rng()
print(g.polyagamma(1, 2))
print(g.polyagamma(1, 2, size=10))

# one can pass an output array
out = np.empty(5)
g.polyagamma(1, 2, out=out)
print(out)

# other numpy distributions are still accessible
print(g.standard_normal())
print(g.standard_gamma())

TODO

  • Add devroye and gamma convolution methods.
  • Add the "alternative" sampling method.
  • Add the "saddle point approximation" method.
  • Add the hybrid sampler based on all four methods.
  • Add array broadcasting support for paramater inputs.

References

  • Polson, Nicholas G., James G. Scott, and Jesse Windle. "Bayesian inference for logistic models using Pólya–Gamma latent variables." Journal of the American statistical Association 108.504 (2013): 1339-1349.
  • J. Windle, N. G. Polson, and J. G. Scott. "Improved Polya-gamma sampling". Technical Report, University of Texas at Austin, 2013b.
  • Windle, Jesse, Nicholas G. Polson, and James G. Scott. "Sampling Polya-Gamma random variates: alternate and approximate techniques." arXiv preprint arXiv:1405.0506 (2014)

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

polyagamma-0.1.0a3.tar.gz (123.1 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page