Corey Morris
commited on
Commit
•
6d41115
1
Parent(s):
9549fcc
Refactor to make later code changes easier
Browse files- result_data_processor.py +8 -2
result_data_processor.py
CHANGED
@@ -60,8 +60,14 @@ class ResultDataProcessor:
|
|
60 |
|
61 |
|
62 |
def process_data(self):
|
63 |
-
|
64 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
65 |
|
66 |
data = pd.concat(dataframes, axis=1).transpose()
|
67 |
|
|
|
60 |
|
61 |
|
62 |
def process_data(self):
|
63 |
+
|
64 |
+
dataframes = []
|
65 |
+
for filename in self._find_files(self.directory, self.pattern):
|
66 |
+
raw_data = self._read_and_transform_data(filename)
|
67 |
+
model_name = filename.split('/')[2]
|
68 |
+
cleaned_data = self._cleanup_dataframe(raw_data, model_name)
|
69 |
+
dataframes.append(cleaned_data)
|
70 |
+
|
71 |
|
72 |
data = pd.concat(dataframes, axis=1).transpose()
|
73 |
|