Diff of /train.py [000000] .. [bb4041]

Switch to side-by-side view

--- a
+++ b/train.py
@@ -0,0 +1,19 @@
+from chatterbot import ChatBot
+from chatterbot.trainers import ListTrainer
+import os
+
+try:
+	os.remove("db.sqlite3")
+	print("Old database removed. Training new database")
+except:
+	print('No database found. Creating new database.')
+
+english_bot = ChatBot('Bot')
+english_bot.set_trainer(ListTrainer)
+for file in os.listdir('data'):
+        print('Training using '+file)
+        convData = open('data/' + file).readlines()
+        english_bot.train(convData)
+        print("Training completed for "+file)
+    
+