CRMArena-Leaderboard / src /populate.py
Kung-Hsiang Huang
update initial lb results
a708f96
raw
history blame
552 Bytes
import os
import pandas as pd
from src.display.utils import AutoEvalColumn
def get_leaderboard_df_crm(
crm_results_path: str, cols: list
) -> tuple[pd.DataFrame, pd.DataFrame]:
"""Creates a dataframe from all the individual experiment results"""
model_performance_df = pd.read_csv(os.path.join(crm_results_path, "all_results.csv"))
model_performance_df = model_performance_df[cols].round(decimals=2)
model_performance_df = model_performance_df.sort_values("Overall ⬆️", ascending=False)
return model_performance_df