def plot_result(result, dirname='./'): # extract variables config = result['config'] forward_diagnostics_data = result['forward_diagnostics_data'] diagnostics_data_list = result['diagnostics_data_list'] processed_data = result['processed_data'] kl_series_list_dict = processed_data['kl_series_list_dict'] # _dirname = eu._generate_dirname(dirname_prefix, 10, config) save_kwargs = dict(dir=os.path.join(dirname, _dirname)) get_tuple = lambda parameter: (parameter, config[parameter]) parameters = dict(map(get_tuple, parameters_to_show)) if 'cctypes' in config: # FIXME: remove this kludgy if statement counter = collections.Counter(config['cctypes']) parameters['Counter(cctypes)'] = dict(six.iteritems(counter)) pass # plot_all_diagnostic_data(forward_diagnostics_data, diagnostics_data_list, kl_series_list_dict, parameters, save_kwargs) return
def plot_result(result, dirname='./'): # extract variables config = result['config'] forward_diagnostics_data = result['forward_diagnostics_data'] diagnostics_data_list = result['diagnostics_data_list'] processed_data = result['processed_data'] kl_series_list_dict = processed_data['kl_series_list_dict'] # _dirname = eu._generate_dirname(dirname_prefix, 10, config) save_kwargs = dict(dir=os.path.join(dirname, _dirname)) get_tuple = lambda parameter: (parameter, config[parameter]) parameters = dict(map(get_tuple, parameters_to_show)) if 'cctypes' in config: # FIXME: remove this kludgy if statement counter = collections.Counter(config['cctypes']) parameters['Counter(cctypes)'] = dict(six.iteritems(counter)) pass # plot_all_diagnostic_data( forward_diagnostics_data, diagnostics_data_list, kl_series_list_dict, parameters, save_kwargs) return
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 pass
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) this_dirname = eru._generate_dirname(dirname_prefix, 10, result['config']) filename_img = os.path.join(dirname_prefix, this_dirname, results_filename+'.png') eu.plot_haystacks_break(result, filename=filename_img) pass else: result = run_experiment(argsdict) if generate_plots: this_dirname = eru._generate_dirname(dirname_prefix, 10, result['config']) filename_img = os.path.join(dirname_prefix, this_dirname, results_filename+'.png') eu.make_folder(os.path.join(dirname_prefix, this_dirname)) eu.plot_haystacks_break(result, filename=filename_img)