SYMBOL INDEX (114 symbols across 12 files) FILE: tests/test_autocorr.py function get_chain (line 6) | def get_chain(seed=42, ndim=5, N=100000): function test_1d (line 15) | def test_1d(seed=42): FILE: tests/test_fwrapper.py function func0 (line 7) | def func0(x): function func1 (line 11) | def func1(x, mu): function func2 (line 15) | def func2(x, mu, ivar): function test_none (line 19) | def test_none(func=func0,seed=42): function test_args1 (line 29) | def test_args1(func=func1,seed=42): function test_args2 (line 40) | def test_args2(func=func2,seed=42): function test_kwargs1 (line 52) | def test_kwargs1(func=func1,seed=42): function test_kwargs2 (line 63) | def test_kwargs2(func=func2,seed=42): function test_argskwargs (line 75) | def test_argskwargs(func=func2,seed=42): FILE: tests/test_sampler.py function logp (line 6) | def logp(x): function test_mean (line 10) | def test_mean(logp=logp,seed=42): function test_std (line 23) | def test_std(logp=logp,seed=42): function test_ncall (line 34) | def test_ncall(seed=42): FILE: tests/test_samples.py function test_chain (line 7) | def test_chain(seed=42): function test_flatten (line 22) | def test_flatten(seed=42): function test_multiple (line 40) | def test_multiple(): FILE: zeus/autocorr.py function _autocorr_func_1d (line 4) | def _autocorr_func_1d(x, norm=True): function _autocorr_time_1d (line 38) | def _autocorr_time_1d(y, c=5.0, method='mk'): function AutoCorrTime (line 80) | def AutoCorrTime(samples, c=5.0, method='mk'): FILE: zeus/callbacks.py class AutocorrelationCallback (line 10) | class AutocorrelationCallback: method __init__ (line 28) | def __init__(self, ncheck=100, dact=0.01, nact=10, discard=0.5, trigge... method __call__ (line 40) | def __call__(self, i, x, y): class SplitRCallback (line 73) | class SplitRCallback: method __init__ (line 87) | def __init__(self, ncheck=100, epsilon=0.05, nsplits=2, discard=0.5, t... method __call__ (line 98) | def __call__(self, i, x, y): method estimate_Rhat (line 137) | def estimate_Rhat(self, means, vars, N): class MinIterCallback (line 157) | class MinIterCallback: method __init__ (line 165) | def __init__(self, nmin=1000): method __call__ (line 168) | def __call__(self, i, x, y): class ParallelSplitRCallback (line 188) | class ParallelSplitRCallback: method __init__ (line 204) | def __init__(self, ncheck=100, epsilon=0.01, nsplits=2, discard=0.5, t... method __call__ (line 220) | def __call__(self, i, x, y): method estimate_Rhat (line 265) | def estimate_Rhat(self, means, vars, N): class SaveProgressCallback (line 285) | class SaveProgressCallback: method __init__ (line 294) | def __init__(self, filename='./chains.h5', ncheck=100): method __call__ (line 304) | def __call__(self, i, x, y): method __save (line 327) | def __save(self, x, y): method __initialize_and_save (line 335) | def __initialize_and_save(self, x, y): FILE: zeus/ensemble.py class EnsembleSampler (line 17) | class EnsembleSampler: method __init__ (line 45) | def __init__(self, method run (line 142) | def run(self, *args, **kwargs): method reset (line 147) | def reset(self): method get_chain (line 154) | def get_chain(self, flat=False, thin=1, discard=0): method get_log_prob (line 177) | def get_log_prob(self, flat=False, thin=1, discard=0): method get_blobs (line 199) | def get_blobs(self, flat=False, thin=1, discard=0): method chain (line 221) | def chain(self): method act (line 232) | def act(self): method ess (line 243) | def ess(self): method ncall (line 254) | def ncall(self): method efficiency (line 265) | def efficiency(self): method scale_factor (line 276) | def scale_factor(self): method get_last_sample (line 286) | def get_last_sample(self): method get_last_sample (line 292) | def get_last_sample(self): method get_last_log_prob (line 299) | def get_last_log_prob(self): method get_last_blobs (line 306) | def get_last_blobs(self): method summary (line 314) | def summary(self): method compute_log_prob (line 333) | def compute_log_prob(self, coords): method run_mcmc (line 388) | def run_mcmc(self, method sample (line 445) | def sample(self, class sampler (line 724) | class sampler(EnsembleSampler): method __init__ (line 725) | def __init__(self, *args, **kwargs): FILE: zeus/fwrapper.py class _FunctionWrapper (line 3) | class _FunctionWrapper(object): method __init__ (line 17) | def __init__(self, f, args, kwargs): method __call__ (line 22) | def __call__(self, x): FILE: zeus/moves.py class DifferentialMove (line 17) | class DifferentialMove: method __init__ (line 33) | def __init__(self, tune=True, mu0=1.0): method get_direction (line 38) | def get_direction(self, X, mu): class GaussianMove (line 66) | class GaussianMove: method __init__ (line 81) | def __init__(self, tune=False, mu0=1.0, cov=None): method get_direction (line 87) | def get_direction(self, X, mu): class GlobalMove (line 118) | class GlobalMove: method __init__ (line 140) | def __init__(self, tune=True, mu0=1.0, rescale_cov=0.001, n_components... method get_direction (line 151) | def get_direction(self, X, mu): class KDEMove (line 192) | class KDEMove: method __init__ (line 210) | def __init__(self, tune=False, mu0=1.0, bw_method=None): method get_direction (line 220) | def get_direction(self, X, mu): class RandomMove (line 250) | class RandomMove: method __init__ (line 266) | def __init__(self, tune=True, mu0=1.0): method get_direction (line 271) | def get_direction(self, X, mu): FILE: zeus/parallel.py function _import_mpi (line 7) | def _import_mpi(use_dill=False): class MPIPool (line 21) | class MPIPool: method __init__ (line 35) | def __init__(self, comm=None): method wait (line 59) | def wait(self): method map (line 81) | def map(self, worker, tasks): method close (line 143) | def close(self): method is_master (line 152) | def is_master(self): method is_worker (line 156) | def is_worker(self): method __enter__ (line 160) | def __enter__(self): method __exit__ (line 164) | def __exit__(self, *args): function split_ranks (line 169) | def split_ranks(N_ranks, N_chunks): class ChainManager (line 189) | class ChainManager: method __init__ (line 204) | def __init__(self, nchains=1, comm=None): method __enter__ (line 220) | def __enter__(self): method __exit__ (line 258) | def __exit__(self, exc_type, exc_value, exc_traceback): method get_rank (line 276) | def get_rank(self): method get_pool (line 284) | def get_pool(self): method gather (line 293) | def gather(self, x, root): method scatter (line 312) | def scatter(self, x, root): method allgather (line 331) | def allgather(self, x): method bcast (line 348) | def bcast(self, x, root): FILE: zeus/plotting.py function cornerplot (line 3) | def cornerplot(samples, function _quantile (line 194) | def _quantile(x, q, weights=None): FILE: zeus/samples.py class samples (line 3) | class samples: method __init__ (line 12) | def __init__(self, ndim, nwalkers): method extend (line 19) | def extend(self, n, blobs): method save (line 43) | def save(self, x, logp, blobs): method chain (line 58) | def chain(self): method length (line 69) | def length(self): method flatten (line 79) | def flatten(self, discard=0, thin=1): method logprob (line 94) | def logprob(self): method flatten_logprob (line 104) | def flatten_logprob(self, discard=0, thin=1): method flatten_blobs (line 118) | def flatten_blobs(self, discard=0, thin=1):