judgerbench/preprocess/generate_table.py

#%%
import pandas as pd

from judgerbench.meta_data import (
    DATADIR
)

#%%
df = pd.read_csv(DATADIR / "summary_20241007_221023.csv")
df

# %%
df_reshaped = (
    df
    .drop(columns=["dataset", "mode", "version"])
    .melt(
        id_vars=["metric"], 
        var_name="model", 
        value_name="value"
    )
    .pivot(index=["model"], columns=["metric"], values='value')
)
df_reshaped.columns.name = None
df_reshaped.reset_index(inplace=True)

df_reshaped
# %%