Readd predict method

JulianKniephoff · May 15, 2016 · 45599d4 · 45599d4
1 parent 098c808
commit 45599d4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/cnn.py b/cnn.py
@@ -193,11 +193,11 @@ def tweet_generator():
             *args, **kwargs
         )
 
-    #def predict(self, tweets, *args, **kwargs):
-    #    return self.network.predict(
-    #        {'input': self.tweets_to_indices(tweets)},
-    #        *args, **kwargs
-    #    )
+    def predict(self, tweets, *args, **kwargs):
+        return self.network.predict(
+            {'input': self.tweets_to_indices(tweets)},
+            *args, **kwargs
+        )
 
     def save(self, basedir):
         makedirs(basedir, exist_ok=True)