| import pandas as pd | |
| def load_dataframe() -> pd.DataFrame: | |
| """ | |
| Load dataframe from the csv file in public directory | |
| Returns | |
| dataframe: a pd.DataFrame of the average scores of the LLMs on each task | |
| """ | |
| dataframe = pd.read_csv("public/datasets/models_scores.csv") | |
| dataframe = dataframe.drop(columns = "Unnamed: 0") | |
| return dataframe | |
| def sort_by(dataframe: pd.DataFrame, column_name: str, ascending:bool = False) -> pd.DataFrame: | |
| """ | |
| Sort the dataframe by column_name | |
| Arguments: | |
| - dataframe: a pandas dataframe to sort | |
| - column_name: a string stating the column to sort the dataframe by | |
| - ascending: a boolean stating to sort in ascending order or not, default to False | |
| Returns: | |
| a sorted dataframe | |
| """ | |
| return dataframe.sort_values(by = column_name, ascending = ascending ) |