def sample_files(model_pkl, data_pkl, optim_pkl, est_pkl): model = read_pkl(model_pkl) data = read_pkl(data_pkl) optim = read_pkl(optim_pkl) fit, est = sample(model, data, optim) write_pkl(est_pkl, est)
def compile_model(stan_fname, pkl_fname): write_pkl(pkl_fname, pystan.StanModel(file=stan_fname))
def optimize_files(model_pkl, data_pkl, out_pkl, n=1): n = int(n) model = read_pkl(model_pkl) data = read_pkl(data_pkl) write_pkl(out_pkl, optimize(model, data, n=n))
import sys import numpy as np from util.pkl import write_pkl, read_pkl data = { 'ct': np.r_[4.7, 4.8], 'cmu': np.r_[-1.5, -1.5], 'csd': np.r_[0.3, 0.4], 'ci': np.r_[1, 3], 'nc': 2, 'nn': 2, } pkl_fname = sys.argv[1] write_pkl(pkl_fname, data)