Spaces:
Paused
Paused
Add aliases to tasks
Browse files
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
|