Shiyu Zhao commited on
Commit
5c7b480
·
1 Parent(s): 3afe0be

Update space

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- 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())
 
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())