Corey Morris
commited on
Commit
•
7d69bda
1
Parent(s):
41d7691
moved organization column to the front
Browse files- 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
|