Exemplo n.º 1
0
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
Exemplo n.º 2
0
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
Exemplo n.º 3
0
    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
Exemplo n.º 4
0
    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)