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)))