Skip to content

A parallel extension of the affine invariant ensemble sampler w/ optional ensemble mixing

Notifications You must be signed in to change notification settings

ZuckermanLab/pyPAM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

74 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

pyPAM

Test Docs

A parallel extension of the affine invariant ensemble sampler emcee w/ optional ensemble mixing

Getting started

Clone repository:

git clone https://github.com/ZuckermanLab/pyPAM.git

Example usage:

import parallel_affine as pa
sampler = pa.ParallelEnsembleSampler(n_ensembles, n_walkers, n_dim, log_prob, log_prob_args, thin, backend_fnames, moves)
states = sampler.run_mixing_sampler(p_0, n_steps_list, n_cores, n_mixing_stages,thin,run_id_list)
pa.plot_single_ensemble_mixing_distributions(sampler)  # quick plot distributions for each ensemble
D = pau.get_data_all_runs(sampler, flat=True)  # get data --> shape = (n ensembles, n mixing stages, n_steps/thin, n_dim)

see Docs for more info.

August George, Zuckerman Lab, OHSU, 2023

About

A parallel extension of the affine invariant ensemble sampler w/ optional ensemble mixing

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published