Corey Morris commited on
Commit
6d41115
1 Parent(s): 9549fcc

Refactor to make later code changes easier

Browse files
Files changed (1) hide show
  1. 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
- dataframes = [self._cleanup_dataframe(self._read_and_transform_data(filename), filename.split('/')[2])
64
- for filename in self._find_files(self.directory, self.pattern)]
 
 
 
 
 
 
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