comparator / src /results.py
albertvillanova's picture
Fix latest_result_path_per_model variable and refactor
c2c9efa verified
raw
history blame
2.92 kB
import json
import gradio as gr
import pandas as pd
from huggingface_hub import HfFileSystem
from src.constants import RESULTS_DATASET_ID, TASKS
def fetch_result_paths():
fs = HfFileSystem()
paths = fs.glob(f"{RESULTS_DATASET_ID}/**/**/*.json")
return paths
def filter_latest_result_path_per_model(paths):
from collections import defaultdict
d = defaultdict(list)
for path in paths:
model_id, _ = path[len(RESULTS_DATASET_ID) + 1:].rsplit("/", 1)
d[model_id].append(path)
return {model_id: max(paths) for model_id, paths in d.items()}
def update_load_results_component():
return gr.Button("Load Results", interactive=True)
def load_results_dataframe(model_id, result_path_per_model=None):
if not model_id or not result_path_per_model:
return
result_path = result_path_per_model[model_id]
fs = HfFileSystem()
with fs.open(result_path, "r") as f:
data = json.load(f)
model_name = data.get("model_name", "Model")
df = pd.json_normalize([{key: value for key, value in data.items()}])
# df.columns = df.columns.str.split(".") # .split return a list instead of a tuple
return df.set_index(pd.Index([model_name])).reset_index()
def load_results_dataframes(*model_ids, result_path_per_model=None):
return [load_results_dataframe(model_id, result_path_per_model=result_path_per_model) for model_id in model_ids]
def display_results(task, *dfs):
dfs = [df.set_index("index") for df in dfs if "index" in df.columns]
if not dfs:
return None, None
df = pd.concat(dfs)
df = df.T.rename_axis(columns=None)
return display_tab("results", df, task), display_tab("configs", df, task)
def display_tab(tab, df, task):
df = df.style.format(na_rep="")
df.hide(
[
row
for row in df.index
if (
not row.startswith(f"{tab}.")
or row.startswith(f"{tab}.leaderboard.")
or row.endswith(".alias")
or (not row.startswith(f"{tab}.{task}") if task != "All" else False)
)
],
axis="index",
)
start = len(f"{tab}.leaderboard_") if task == "All" else len(f"{tab}.{task} ")
df.format_index(lambda idx: idx[start:].removesuffix(",none"), axis="index")
return df.to_html()
def update_tasks_component():
return gr.Radio(
["All"] + list(TASKS.values()),
label="Tasks",
info="Evaluation tasks to be displayed",
value="All",
interactive=True,
)
def clear_results():
# model_id_1, model_id_2, dataframe_1, dataframe_2, task
return (
None, None, None, None,
gr.Radio(
["All"] + list(TASKS.values()),
label="Tasks",
info="Evaluation tasks to be displayed",
value="All",
interactive=False,
),
)