Moves _preprocess_data up to Classifier class

snorkel-team · Dec 29, 2017 · 78655e7 · 78655e7
1 parent 78aac45
commit 78655e7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/snorkel/learning/classifier.py b/snorkel/learning/classifier.py
@@ -121,6 +121,10 @@ def error_analysis(self, session, X_test, Y_test,
         return s.score(test_marginals, train_marginals=None, b=b,
             display=display, set_unlabeled_as_neg=set_unlabeled_as_neg)
 
+    def _preprocess_data(self, X):
+        """Generic preprocessing subclass; may be called by external methods."""
+        return X
+
     def save(self):
         raise NotImplementedError()
 

diff --git a/snorkel/learning/disc_learning.py b/snorkel/learning/disc_learning.py
@@ -345,7 +345,3 @@ def load(self, model_name=None, save_dir='checkpoints', verbose=True):
         else:
             raise Exception("[{0}] No model found at <{1}>".format(
                 self.name, model_name))
-
-    def _preprocess_data(self, X):
-        """Generic preprocessing subclass; may be called by external methods."""
-        return X