Initialize stemming method

gunthercox · Nov 3, 2018 · 205a3a8 · 205a3a8
1 parent ed166ed
commit 205a3a8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/chatterbot/stemming.py b/chatterbot/stemming.py
@@ -16,6 +16,15 @@ def __init__(self, language='english'):
         self.stopwords = nltk.corpus.stopwords.words(language)
         self.stopwords.append('')
 
+    def initialize(self):
+        """
+        Initialize the stemmer.
+        """
+        from chatterbot.utils import nltk_download_corpus
+
+        nltk.download('stopwords')
+        # nltk_download_corpus('stopwords')
+
     def stem(self, text):
 
         if not text:

diff --git a/chatterbot/storage/storage_adapter.py b/chatterbot/storage/storage_adapter.py
@@ -21,6 +21,8 @@ def __init__(self, base_query=None, *args, **kwargs):
             'stemmer_language', 'english'
         ))
 
+        self.stemmer.initialize()
+
     def get_model(self, model_name):
         """
         Return the model class for a given model name.