def main():
dataset = dataset = CornellMovieDialogs(data_directory="data/", vocabulary_size=20000)
if not os.path.exists("data/cleaned"):
dataset.make_conversations()
else:
dataset.load()
bot = Chatbot(sequence_length=10, hidden_size=128, vocabulary_size=dataset.vocabulary_size)
# bot.load("models/checkpoints/checkpoint.h5")
conversation_generator = gen(dataset, bot.sequence_length, batch_size=10, max_samples=5000) #gen(dataset, bot.sequence_length, batch_size=10) #
bot.fit_generator(conversation_generator, samples_per_epoch=10000, nb_epoch=100)
bot.save("models/chatbot.h5")
