plotting.py 文件源码

def show_comparison(expt_base, algorithms, subset, ymin=None, ymax=None):
    expt_names = ['/'.join([expt_base, alg]) for alg in algorithms]
    labels = [ALGORITHM_LABELS[alg] for alg in algorithms]
    plot_log_probs(expt_names, subset=subset, labels=labels)
    pylab.xlim(1e1, 1e5)
    if ymin is None:
        ymin, ymax = get_ylim(expt_base)
    pylab.ylim(ymin, ymax)
    pylab.xlabel('Wall clock time (seconds)', fontsize='x-large')
    pylab.ylabel('{} log-probabilities'.format({'train': 'Training', 'test': 'Test'}[subset]),
                 fontsize='x-large')
    pylab.tight_layout()