Spaces:
Running
Running
natolambert
commited on
Commit
•
4b6a00c
1
Parent(s):
35e2ca1
update splits
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def avg_over_herm(dataframe):
|
|
47 |
# for each subset, avg the columns that have the subset in the column name, then add a new column with subset name and avg
|
48 |
for subset in subsets:
|
49 |
if subset == "refusals":
|
50 |
-
subset_cols = ["refusals-dangerous", "refusals-offensive", "donotanswer","xstest"]
|
51 |
else:
|
52 |
subset_cols = [col for col in new_df.columns if subset in col]
|
53 |
new_df[subset] = np.round(np.nanmean(new_df[subset_cols].values, axis=1), 2)
|
|
|
47 |
# for each subset, avg the columns that have the subset in the column name, then add a new column with subset name and avg
|
48 |
for subset in subsets:
|
49 |
if subset == "refusals":
|
50 |
+
subset_cols = ["refusals-dangerous", "refusals-offensive", "donotanswer","xstest-should-refuse", "xstest-shouldnt-refuse"]
|
51 |
else:
|
52 |
subset_cols = [col for col in new_df.columns if subset in col]
|
53 |
new_df[subset] = np.round(np.nanmean(new_df[subset_cols].values, axis=1), 2)
|