from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
from rasa_core.agent import Agent
from rasa_core.policies.keras_policy import KerasPolicy
from rasa_core.policies.memoization import MemoizationPolicy
if __name__ == '__main__':
agent = Agent("domain.yml",
policies=[MemoizationPolicy(), KerasPolicy()])
agent.visualize("data/stories.md",
output_file="graph.png", max_history=2)
help="path of the Rasa NLU training data, "
"used to insert example messages into the graph")
utils.add_logging_option_arguments(parser)
return parser
if __name__ == '__main__':
parser = create_argument_parser()
args = parser.parse_args()
utils.configure_colored_logging(args.loglevel)
agent = Agent(args.domain, policies=[MemoizationPolicy(), KerasPolicy()])
# this is optional, only needed if the `_greet` type of
# messages in the stories should be replaced with actual
# messages (e.g. `hello`)
if args.nlu_data is not None:
from rasa_nlu.training_data import load_data
nlu_data = load_data(args.nlu_data)
else:
nlu_data = None
logger.info("Starting to visualize stories...")
agent.visualize(args.stories, args.output, args.max_history,
nlu_training_data=nlu_data)
logger.info("Finished graph creation. Saved into {}".format(
os.path.abspath(args.output)))
# -*- coding: utf-8 -*-
# ** Project : rasa_demo
# ** Created by: Yizhen
# ** Date: 2019/7/2
# ** Time: 16:24
"""
method 1:
直接执行
python -m rasa_core.visualize -d config/chat_domain.yml -s data/stories.md -o graph.jpg
"""
import logging
from rasa_core.agent import Agent
from rasa_core.policies import MemoizationPolicy, KerasPolicy
logger = logging.getLogger(__name__)
agent = Agent('config/chat_domain.yml',
policies=[MemoizationPolicy(),
KerasPolicy()])
nlu_data = None
logger.info("Starting to visualize stories...")
agent.visualize('data/stories.md', 'graph.jpg', 10, nlu_training_data=nlu_data)
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
from rasa_core.agent import Agent
from rasa_core.policies.keras_policy import KerasPolicy
from rasa_core.policies.memoization import MemoizationPolicy
if __name__ == '__main__':
agent = Agent("chat_bot_domain.yml",
policies=[MemoizationPolicy(),
KerasPolicy()])
agent.visualize("data/stories.md", output_file="graph.png", max_history=2)
from IPython.display import Image
from rasa_core.agent import Agent
agent = Agent('domain.yml')
agent.visualize("stories.md", "story_graph.png", max_history=2)
Image(filename="story_graph.png")
from rasa_core.agent import Agent
from rasa_core.policies.keras_policy import KerasPolicy
from rasa_core.policies.memoization import MemoizationPolicy
if __name__ == '__main__':
agent = Agent("./rasa_core/domain.yml",
policies=[MemoizationPolicy(),
KerasPolicy()])
agent.visualize("./rasa_core/stories.md",
output_file="graph.png",
max_history=5)
"used to insert example messages into the graph")
utils.add_logging_option_arguments(parser)
return parser
if __name__ == '__main__':
arg_parser = create_argument_parser()
args = arg_parser.parse_args()
utils.configure_colored_logging(args.loglevel)
agent = Agent(args.domain, policies=[MemoizationPolicy(), KerasPolicy()])
# this is optional, only needed if the `_greet` type of
# messages in the stories should be replaced with actual
# messages (e.g. `hello`)
if args.nlu_data is not None:
from rasa_nlu.training_data import load_data
nlu_data = load_data(args.nlu_data)
else:
nlu_data = None
logger.info("Starting to visualize stories...")
agent.visualize(args.stories, args.output, args.max_history,
nlu_training_data=nlu_data)
logger.info("Finished graph creation. Saved into {}".format(
os.path.abspath(args.output)))
