def extract_sensitivity_results()

in figs/fig_S8.py [0:0]

• 21 lines of code
• 8 McCabe index (conditional complexity)

def extract_sensitivity_results():
    res = {}
    for fname in [
        'branin_gramacy_100',
        'sensitivity',
    ]:
        with open(f'../benchmarks/results/{fname}_aggregated_results.json', 'r') as fin:
            res.update(object_from_json(json.load(fin)))

    # Results for D=100
    ys1 = {}
    for d in [2, 3, 4, 5, 6, 7, 8]:
        if d == 4:
            ys1[d] = res['Branin, D=100'].objective_at_true_best['ALEBO']
        else:
            ys1[d] = res['Branin, D=100_sensitivity'].objective_at_true_best[f'ALEBO, d={d}']

    # Results for d_e=4
    ys2 = {}
    for D in [50, 100, 200, 500, 1000]:
        if D == 100:
            ys2[D] = res['Branin, D=100'].objective_at_true_best['ALEBO']
        else:
            ys2[D] = res[f'Branin, D={D}_sensitivity'].objective_at_true_best['ALEBO']
    return ys1, ys2