Skip to main content

HDDM is a python module that implements Hierarchical Bayesian estimation of Drift Diffusion Models.

Project description

Author:

Thomas V. Wiecki, Imri Sofer, Michael J. Frank

Contact:
thomas_wiecki@brown.edu, imri_sofer@brown.edu, michael_frank@brown.edu
Web site:

http://ski.clps.brown.edu/hddm_docs

Github:

http://github.com/hddm-devs/hddm

Mailing list:

https://groups.google.com/group/hddm-users/

License:

HDDM is released under the BSD 2 license.

Version:
0.5.3
https://secure.travis-ci.org/hddm-devs/hddm.png?branch=develop

Purpose

HDDM is a python toolbox for hierarchical Bayesian parameter estimation of the Drift Diffusion Model (via PyMC). Drift Diffusion Models are used widely in psychology and cognitive neuroscience to study decision making.

Features

  • Uses hierarchical Bayesian estimation (via PyMC) of DDM parameters to allow simultaneous estimation of subject and group parameters, where individual subjects are assumed to be drawn from a group distribution. HDDM should thus produce better estimates when less RT values are measured compared to other methods using maximum likelihood for individual subjects (i.e. DMAT or fast-dm).

  • Heavily optimized likelihood functions for speed (Navarro & Fuss, 2009).

  • Flexible creation of complex models tailored to specific hypotheses (e.g. estimation of separate drift-rates for different task conditions; or predicted changes in model parameters as a function of other indicators like brain activity).

  • Estimate trial-by-trial correlations between a brain measure (e.g. fMRI BOLD) and a diffusion model parameter using the HDDMRegression model.

  • Built-in Bayesian hypothesis testing and several convergence and goodness-of-fit diagnostics.

Quick-start

The following is a minimal python script to load data, run a model and examine its parameters and fit.

import hddm

# Load data from csv file into a NumPy structured array
data = hddm.load_csv('simple_difficulty.csv')

# Create a HDDM model multi object
model = hddm.HDDM(data, depends_on={'v':'difficulty'})

# Create model and start MCMC sampling
model.sample(2000, burn=20)

# Print fitted parameters and other model statistics
model.print_stats()

# Plot posterior distributions and theoretical RT distributions
model.plot_posteriors()
model.plot_posterior_predictive()

For more information about the software and theories behind it, please see the main publication.

Installation

The easiest way to install HDDM is through Anaconda (available for Windows, Linux and OSX):

  1. Download and install Anaconda.

  2. In a shell (Windows: Go to Start->Programs->Anaconda->Anaconda command prompt) type:

conda install -c pymc pymc
conda install -c twiecki hddm

Note that you only have to do this once. When a new HDDM version is available, conda update will automatically update HDDM.

If you want to use pip instead of conda, type:

pip install pandas
pip install pymc
pip install kabuki
pip install hddm

This might require super-user rights via sudo. Note that this installation method is discouraged as it leads to all kinds of problems on various platforms.

If you are having installation problems please contact the mailing list.

How to cite

If HDDM was used in your research, please cite the publication:

Wiecki TV, Sofer I and Frank MJ (2013). HDDM: Hierarchical Bayesian estimation of the Drift-Diffusion Model in Python. Front. Neuroinform. 7:14. doi: 10.3389/fninf.2013.00014

Getting started

Check out the tutorial on how to get started. Further information can be found in howto and the documentation.

Join our low-traffic mailing list.

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

HDDM-0.5.3.tar.gz (993.1 kB view details)

Uploaded Source

File details

Details for the file HDDM-0.5.3.tar.gz.

File metadata

  • Download URL: HDDM-0.5.3.tar.gz
  • Upload date:
  • Size: 993.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No

File hashes

Hashes for HDDM-0.5.3.tar.gz
Algorithm Hash digest
SHA256 345945206e40f44813a1c8c3144a24672b666cb724273b131088011096a0d2d7
MD5 ebc47f4bdda72af04647a0650030b7fb
BLAKE2b-256 d1c98d6a9be98d49e7fe5f7eed3b971ef34e87f2e2939b089d34830532a00e91

See more details on using hashes here.

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