-
Notifications
You must be signed in to change notification settings - Fork 0
/
train_online.py
40 lines (32 loc) · 1.36 KB
/
train_online.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
#! /usr/local/bin/python3.6
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
import logging
from rasa_core.channels.console import ConsoleInputChannel
from rasa_core.agent import Agent
from rasa_core.interpreter import RegexInterpreter
from rasa_core.policies.keras_policy import KerasPolicy
from rasa_core.policies.memoization import MemoizationPolicy
from rasa_core.interpreter import RasaNLUInterpreter
logger = logging.getLogger(__name__)
def run_news_online(input_channel,
interpreter,
domain_def_file='./domain/domain.yml',
training_data_file='./data/stories.md',
):
agent = Agent(domain_def_file,
policies=[MemoizationPolicy(max_history=3), KerasPolicy()],
interpreter=interpreter)
training_data = agent.load_data(training_data_file)
agent.train_online(training_data,
input_channel=input_channel,
batch_size=50,
epochs=200,
max_training_samples=300)
return agent
if __name__ == '__main__':
logging.basicConfig(level='INFO')
nlu_interpreter = RasaNLUInterpreter('models/tour_guide/default/nlu')
run_news_online(ConsoleInputChannel(), nlu_interpreter)