Shiyu Zhao
commited on
Commit
·
5c7b480
1
Parent(s):
3afe0be
Update space
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ def format_dataframe(df):
|
|
32 |
pivoted_df = pivoted_df.sort_values(['Dataset', 'MRR'], ascending=[True, False])
|
33 |
|
34 |
# Reorder columns
|
35 |
-
column_order = ['Method', 'Hit@1', 'Hit@5', 'R@20', 'MRR']
|
36 |
pivoted_df = pivoted_df.reindex(columns=column_order)
|
37 |
|
38 |
return pivoted_df
|
@@ -41,7 +41,7 @@ def create_leaderboard(dataset):
|
|
41 |
formatted_df = format_dataframe(df)
|
42 |
if dataset != 'All':
|
43 |
formatted_df = formatted_df[formatted_df['Dataset'] == dataset]
|
44 |
-
|
45 |
return formatted_df
|
46 |
|
47 |
datasets = ['All'] + sorted(df.columns[1:].str.split('_').str[0].unique().tolist())
|
|
|
32 |
pivoted_df = pivoted_df.sort_values(['Dataset', 'MRR'], ascending=[True, False])
|
33 |
|
34 |
# Reorder columns
|
35 |
+
column_order = ['Method', 'Dataset', 'Hit@1', 'Hit@5', 'R@20', 'MRR']
|
36 |
pivoted_df = pivoted_df.reindex(columns=column_order)
|
37 |
|
38 |
return pivoted_df
|
|
|
41 |
formatted_df = format_dataframe(df)
|
42 |
if dataset != 'All':
|
43 |
formatted_df = formatted_df[formatted_df['Dataset'] == dataset]
|
44 |
+
formatted_df = formatted_df.drop('Dataset', axis=1)
|
45 |
return formatted_df
|
46 |
|
47 |
datasets = ['All'] + sorted(df.columns[1:].str.split('_').str[0].unique().tolist())
|