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
|