IDEA-Bench-Arena / arena_elo /simple_test.py
JasiLiang's picture
initial commit
62d106f verified
import pickle
with open("./results/latest/elo_results.pkl",'rb') as f:
data = pickle.load(f)
print()
df = data["anony"]["leaderboard_table_df"]
# sort by rating
df = df.sort_values(by=["rating"], ascending=False)
print(df)
print()
df = data["full"]["leaderboard_table_df"]
# sort by rating
df = df.sort_values(by=["rating"], ascending=False)
print(df)
print('done')