def build_df()

in scripts/plot_results.py [0:0]

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

def build_df(dirname):
    df = pd.DataFrame(columns=['n_envs', 'dim_inv', 'dim_spu', 'dataset', 'model', 'mean', 'std'])
    for filename in glob.glob(os.path.join(dirname, "*.jsonl")):
        with open(filename) as f:
            dic = json.load(f)
            n_envs = dic["n_envs"]
            dim_inv = dic["dim_inv"]
            dim_spu = dic["dim_spu"]
            for dataset in dic["data"].keys():
                single_dic = {}
                for model in dic["data"][dataset].keys():
                    mean =  dic["data"][dataset][model]["mean"]
                    std = dic["data"][dataset][model]["std"]
                    single_dic = dict(
                        n_envs=n_envs,
                        dim_inv=dim_inv,
                        dim_spu=dim_spu,
                        dataset=dataset,
                        model=model,
                        mean=mean,
                        std=std
                        )
                    # print(single_dic)
                    df = df.append(single_dic, ignore_index=True)

    return df