def debug_ll(pc, expt, separate, voting, qs_file, indivs_file):
data_params, inference_params, details = specific_params(separate, voting,
use_meta=True,
know=False)
data = wrapper.get_data(pc, expt, data_params)
data = wrapper.process_data(data, inference_params)
return data, calc_ll(qs_file, indivs_file, data, inference_params)
def mh_wrapper(pc, expt, data_params, inference_params, return_stats=False,
results_folder='./', log_folder='./', specified_start=None):
print(datetime.datetime.now())
print_save_params(data_params, inference_params, results_folder)
data = wrapper.get_data(pc, expt, data_params)
data = wrapper.process_data(data, inference_params)
1/0
if inference_params.meta_noise_type == "beta":
data['metas'] = [np.clip(m, .01, .99) for m in data['metas']]
if expt == "generate":
data.save_all(results_folder)
dill.dump(data, open(results_folder + 'data.pkl', 'w'))
if pc == "mac":
parallel = False
else:
parallel = True
lview = setup_parallel(parallel)
if inference_params.separate:
init_state = wrapper.make_init_for_mh(inference_params.vars_to_init,
data, specified_start)
q_results = wrapper.mh_separate(data['beliefs'], data['metas'],
inference_params, lview, init_state,
results_folder, log_folder, return_stats)
indiv_results = None
else:
if inference_params.num_chains > 1:
sys.exit("Multiple chains not implemented for non-separate qs")
results = wrapper.mh_across(data['beliefs'], data['metas'],
inference_params, lview, results_folder,
log_folder, return_stats)
q_results, indiv_results = results
print(datetime.datetime.now())
return q_results, indiv_results, data
