Skip to main content

Python interface for hBayesDM, hierarchical Bayesian modeling of RL-DM tasks

Project description

This is the Python version of hBayesDM (hierarchical Bayesian modeling of Decision-Making tasks), a user-friendly package that offers hierarchical Bayesian analysis of various computational models on an array of decision-making tasks. hBayesDM in Python uses PyStan (Python interface for Stan) for Bayesian inference.

It supports Python 3.5 or higher versions and requires several packages including: NumPy, SciPy, Pandas, PyStan, Matplotlib, and ArviZ.


You can install hBayesDM from PyPI with the following line:

pip install hbayesdm  # Install using pip

If you want to install from source (by cloning from GitHub):

git clone
cd hBayesDM
cd Python

python install  # Install from source


If you used hBayesDM or some of its codes for your research, please cite this paper:

  title = {Revealing Neurocomputational Mechanisms of Reinforcement Learning and Decision-Making With the {hBayesDM} Package},
  author = {Ahn, Woo-Young and Haines, Nathaniel and Zhang, Lei},
  journal = {Computational Psychiatry},
  year = {2017},
  volume = {1},
  pages = {24--57},
  publisher = {MIT Press},
  url = {doi:10.1162/CPSY_a_00002},

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

hbayesdm-1.1.1a1.tar.gz (387.8 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