RELOADBOARD / src /reload.py
Koshti10's picture
Upload 8 files
f5ad77e verified
# Isolated functions to reload the leaderboard data and plot the results
from src.leaderboard_utils import filter_search, get_github_data
from src.plot_utils import split_models, compare_plots
def get_primary_leaderboard():
'''
Returns
primary_leaderboard_df[0]: Dataframe containing the primary leaderboard (laterst version of the benchmark results)
'''
print("Initializing Reload...........")
primary_leaderboard_df, _, _ = get_github_data()
print(primary_leaderboard_df)
return primary_leaderboard_df[0]
def get_open_models():
'''
Returns
open_models: Checkbox group containing the open models
'''
primary_leaderboard_df, _, _ = get_github_data()
temp_df = primary_leaderboard_df[0]
models = list(temp_df[list(temp_df.columns)[0]].unique())
open_models, _ = split_models(models)
return open_models
def get_closed_models():
'''
Returns
closed_models: Checkbox group containing the closed models
'''
primary_leaderboard_df, _, _ = get_github_data()
temp_df = primary_leaderboard_df[0]
models = list(temp_df[list(temp_df.columns)[0]].unique())
_, closed_models = split_models(models)
return closed_models
def get_plot_df():
'''
Returns
plot_df: Dataframe containing the results of latest version for plotting
'''
primary_leaderboard_df, _, _ = get_github_data()
plot_df = primary_leaderboard_df[0]
return plot_df
def get_version_names():
'''
Returns
version_names: List containing the versions of the benchmark results for dropdown selection
'''
_, _, version_names = get_github_data()
return version_names
def get_version_df():
'''
Returns
version_dfs: Dataframe containing the benchmark results for all versions
'''
_, version_dfs, _ = get_github_data()
return version_dfs
def get_prev_df(name='initial'):
'''
Returns
prev_df: Dataframe containing the benchmark results for the previous versions (default = latest version)
'''
_, version_dfs, version_names = get_github_data()
if name == 'initial':
name = version_names[0]
ind = version_names.index(name)
prev_df = version_dfs[ind]
return prev_df