from future import absolute_import from future import division from future import unicode_literals
import logging
from rasa_core.agent import Agent from rasa_core.policies.keras_policy import KerasPolicy from rasa_core.policies.memoization import MemoizationPolicy
if name == ‘main’: logging.basicConfig(level=‘INFO’)
training_data_file = './data/stories.md'
model_path = './models/dialogue'
agent = Agent('weather_domain.yml', policies = [MemoizationPolicy(max_history = 5), KerasPolicy(featurizer)])
agent.train(
training_data_file,
augmentation_factor = 50,
max_history = 5,
epochs = 500,
batch_size = 10,
validation_split = 0.2)
agent.persist(model_path)