Postmodel fixes (#604)

- Add postmodel requirements to requirement/main [Resolves #602] - Change matrixstore references in postmodel.model_evaluator [Resolves #603]
dssg · Feb 20, 2019 · 40c4f98 · 40c4f98
1 parent c003c87
commit 40c4f98
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/requirement/main.txt b/requirement/main.txt
@@ -21,3 +21,6 @@ wrapt==1.11.1
 argcmdr==0.6.0
 sqlparse==0.2.4
 pebble==4.3.9
+seaborn==0.9.0
+adjustText==0.7.3
+graphviz==0.10.1
diff --git a/src/triage/component/postmodeling/contrast/model_evaluator.py b/src/triage/component/postmodeling/contrast/model_evaluator.py
@@ -302,7 +302,7 @@ def preds_matrix(self, path):
 
         storage = ProjectStorage(path)
         matrix_storage = MatrixStorageEngine(storage).get_store(self.pred_matrix_uuid)
-        mat = matrix_storage.matrix
+        mat = matrix_storage.design_matrix
 
         # Merge with predictions table and return complete matrix
         merged_df = pd.merge(mat,
@@ -336,7 +336,7 @@ def train_matrix(self, path):
 
         storage = ProjectStorage(path)
         matrix_storage = MatrixStorageEngine(storage).get_store(self.train_matrix_uuid)
-        mat = matrix_storage.matrix
+        mat = matrix_storage.design_matrix
 
         # Merge with predictions table and return complete matrix
         merged_df = mat.merge(self.predictions,