parser.add_argument('--num_iters', default=200, type=int) parser.add_argument('--num_chains', default=1, type=int) parser.add_argument('--ct_kernel', default=0, type=int) # 0 for Gibbs, 1 for MH parser.add_argument('--no_plots', action='store_true') return parser if __name__ == "__main__": import argparse import experiment_runner.experiment_utils as eru from experiment_runner.ExperimentRunner import ExperimentRunner, propagate_to_s3 parser = gen_parser() args = parser.parse_args() argsdict = eu.parser_args_to_dict(args) generate_plots = not argsdict['no_plots'] results_filename = 'recovers_original_densities_results' dirname_prefix = 'recovers_original_densities' er = ExperimentRunner(run_experiment, dirname_prefix=dirname_prefix, bucket_str='experiment_runner', storage_type='fs') er.do_experiments([argsdict]) if generate_plots: for id in er.frame.index: result = er._get_result(id) this_dirname = eru._generate_dirname(dirname_prefix, 10, result['config']) filename_img = os.path.join(dirname_prefix, this_dirname, results_filename+'.png') eu.plot_recovers_original_densities(result, filename=filename_img) pass
parser.add_argument('--no_runner', action='store_true') # do not use experiment runner return parser if __name__ == "__main__": # python haystacks_break.py --num_rows 30 --max_cols 64 --num_chains 8 --rho .9 --save_pickle import argparse import experiment_runner.experiment_utils as eru from experiment_runner.ExperimentRunner import ExperimentRunner, propagate_to_s3 parser = gen_parser() args = parser.parse_args() argsdict = eu.parser_args_to_dict(args) generate_plots = not argsdict['no_plots'] use_runner = not argsdict['no_runner'] results_filename = 'haystacks_break_results' dirname_prefix = 'haystacks_break' er = ExperimentRunner(run_experiment, dirname_prefix=dirname_prefix, bucket_str='experiment_runner', storage_type='fs') if use_runner: er.do_experiments([argsdict], do_multiprocessing=False) if generate_plots: for id in er.frame.index: result = er._get_result(id)