def plot_experiment()

in code/experiment_synthetic/plot.py [0:0]

• 24 lines of code
• 5 McCabe index (conditional complexity)

def plot_experiment(all_solutions, category, fname):
    plt.rcParams["font.family"] = "serif"
    plt.rc('text', usetex=True)
    plt.rc('font', size=10)

    results = {}

    for line in all_solutions:
        words = line.split(" ")
        setup = str(words[0])
        model = str(words[1])
        err_causal = float(words[-2])
        err_noncausal = float(words[-1])

        if setup not in results:
            results[setup] = {}

        if model not in results[setup]:
            results[setup][model] = []

        results[setup][model].append([err_causal, err_noncausal])

    plt.figure(figsize=(7, 2))
    plot_bars(results, category, "causal")
    plot_bars(results, category, "noncausal")
    plt.tight_layout(0, 0, 0.5)

    if fname is None:
        plt.show()
    else:
        plt.savefig(fname)