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
|