natolambert commited on
Commit
4b6a00c
1 Parent(s): 35e2ca1

update splits

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