Skip to content

Instantly share code, notes, and snippets.

@nathaniel-lemonade
Created May 7, 2018 18:47
Show Gist options
  • Save nathaniel-lemonade/80057f7c268f986ea72566ec6391c68c to your computer and use it in GitHub Desktop.
Save nathaniel-lemonade/80057f7c268f986ea72566ec6391c68c to your computer and use it in GitHub Desktop.
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
# from rasa_nlu.converters import load_data
from rasa_nlu.training_data import load_data
from rasa_nlu.config import RasaNLUModelConfig
#from rasa_nlu.config import RasaNLUConfig
from rasa_nlu.model import Trainer, Metadata, Interpreter
from rasa_nlu import config
def train (data, config_file, model_dir):
training_data = load_data(data)
configuration = config.load(config_file)
trainer = Trainer(configuration)
trainer.train(training_data)
model_directory = trainer.persist(model_dir, fixed_model_name = 'chat')
def run():
interpreter = Interpreter.load('./models/nlu/default/chat')
print(interpreter.parse('I want to order pizza'))
#print(interpreter.parse(u'What is the reivew for the movie Die Hard?'))
if __name__ == '__main__':
train('./data/training_data.json', './config/config.yml', './models/nlu')
#run()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment