From 3abbd40b262b05272bed40bbe803f0b1e7906026 Mon Sep 17 00:00:00 2001 From: kenkeiras Date: Sat, 30 Sep 2017 00:53:42 +0200 Subject: [PATCH] Add show_sample visualization. --- naive-nlu/tree_nlu/cli.py | 8 +++++++- naive-nlu/tree_nlu/visualization.py | 5 +++++ 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/naive-nlu/tree_nlu/cli.py b/naive-nlu/tree_nlu/cli.py index 20b53e3..a4eee32 100644 --- a/naive-nlu/tree_nlu/cli.py +++ b/naive-nlu/tree_nlu/cli.py @@ -1,6 +1,9 @@ import logging from .knowledge_base import KnowledgeBase -from .visualization import show_knowledge +from .visualization import ( + show_knowledge, + show_samples, +) from .tests import gac_100 from .modifiable_property import ( ModifiableProperty, @@ -31,6 +34,9 @@ def main(): if data == '/show': show_knowledge(knowledge) continue + elif data == '/samples': + show_samples(knowledge) + continue ret = knowledge.process(data) if ret: diff --git a/naive-nlu/tree_nlu/visualization.py b/naive-nlu/tree_nlu/visualization.py index 4a95a55..6f07325 100644 --- a/naive-nlu/tree_nlu/visualization.py +++ b/naive-nlu/tree_nlu/visualization.py @@ -1,3 +1,8 @@ def show_knowledge(knowledge): for key in knowledge.knowledge: print("\x1b[1m{}\x1b[0m {}".format(key, knowledge.knowledge[key])) + + +def show_samples(knowledge): + for example in knowledge.originals: + print("{}".format(example))