Dont need to store entire statement history

gunthercox · Jun 30, 2017 · 002030a · 002030a
1 parent f1fa616
commit 002030a
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/chatterbot/trainers.py b/chatterbot/trainers.py
@@ -76,17 +76,17 @@ def train(self, conversation):
         Train the chat bot based on the provided list of
         statements that represents a single conversation.
         """
-        statement_history = []
+        previous_statement_text = None
 
         for text in conversation:
             statement = self.get_or_create(text)
 
-            if statement_history:
+            if previous_statement_text:
                 statement.add_response(
-                    Response(statement_history[-1].text)
+                    Response(previous_statement_text)
                 )
 
-            statement_history.append(statement)
+            previous_statement_text = statement.text
             self.storage.update(statement)
 
 
@@ -116,17 +116,17 @@ def train(self, *corpus_paths):
 
             for corpus in corpora:
                 for conversation in corpus:
-                    statement_history = []
+                    previous_statement_text = None
 
                     for text in conversation:
                         statement = self.get_or_create(text)
 
-                        if statement_history:
+                        if previous_statement_text:
                             statement.add_response(
-                                Response(statement_history[-1].text)
+                                Response(previous_statement_text)
                             )
 
-                        statement_history.append(statement)
+                        previous_statement_text = statement.text
                         self.storage.update(statement)