steffenc commited on
Commit
41cd888
·
1 Parent(s): f1d9de3

Add aliases to tasks

Browse files
Files changed (1) hide show
  1. opendashboards/assets/inspect.py +7 -0
opendashboards/assets/inspect.py CHANGED
@@ -4,6 +4,13 @@ import pandas as pd
4
  import opendashboards.utils.utils as utils
5
 
6
  def clean_data(df):
 
 
 
 
 
 
 
7
  return df.dropna(subset=df.filter(regex='completions|rewards').columns, how='any')
8
 
9
  @st.cache_data
 
4
  import opendashboards.utils.utils as utils
5
 
6
  def clean_data(df):
7
+ task_map = {
8
+ 'question-answering': 'QA',
9
+ 'summarization': 'Summarization',
10
+ 'date-based question answering': 'Date QA',
11
+ 'math': 'Math'
12
+ }
13
+ df['task'] = df.task.map(task_map)
14
  return df.dropna(subset=df.filter(regex='completions|rewards').columns, how='any')
15
 
16
  @st.cache_data