Corey Morris commited on
Commit
7d69bda
1 Parent(s): 41d7691

moved organization column to the front

Browse files
Files changed (1) hide show
  1. result_data_processor.py +9 -0
result_data_processor.py CHANGED
@@ -140,6 +140,9 @@ class ResultDataProcessor:
140
  cols = cols[:2] + cols[-1:] + cols[2:-1]
141
  data = data[cols]
142
 
 
 
 
143
  # Drop specific columns
144
  data = data.drop(columns=['all', 'truthfulqa:mc|0'])
145
 
@@ -149,6 +152,12 @@ class ResultDataProcessor:
149
  # move the parameters column to the front of the dataframe
150
  cols = data.columns.tolist()
151
  cols = cols[-1:] + cols[:-1]
 
 
 
 
 
 
152
  data = data[cols]
153
 
154
  # remove extreme outliers from column harness|truthfulqa:mc1
 
140
  cols = cols[:2] + cols[-1:] + cols[2:-1]
141
  data = data[cols]
142
 
143
+
144
+
145
+
146
  # Drop specific columns
147
  data = data.drop(columns=['all', 'truthfulqa:mc|0'])
148
 
 
152
  # move the parameters column to the front of the dataframe
153
  cols = data.columns.tolist()
154
  cols = cols[-1:] + cols[:-1]
155
+ print(cols)
156
+ data = data[cols]
157
+
158
+ # Reorder columns to move 'organization' to the second position
159
+ cols = data.columns.tolist()
160
+ cols = cols[-1:] + cols[:-1]
161
  data = data[cols]
162
 
163
  # remove extreme outliers from column harness|truthfulqa:mc1