From 2c36dd9b7e41178ac47ea5dd8631242b5242b92c Mon Sep 17 00:00:00 2001 From: kenkeiras Date: Sat, 30 Sep 2017 00:55:42 +0200 Subject: [PATCH] Save original examples (needed on show_samples). --- naive-nlu/tree_nlu/knowledge_base.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/naive-nlu/tree_nlu/knowledge_base.py b/naive-nlu/tree_nlu/knowledge_base.py index dfb2b51..ead4b07 100644 --- a/naive-nlu/tree_nlu/knowledge_base.py +++ b/naive-nlu/tree_nlu/knowledge_base.py @@ -15,6 +15,7 @@ def diff_knowledge(before, after): class KnowledgeBase(object): def __init__(self, knowledge, examples=[], trained=[]): self.knowledge = copy.copy(knowledge) + self.originals = [] self.examples = copy.copy(examples) self.trained = copy.copy(trained) @@ -43,6 +44,7 @@ class KnowledgeBase(object): self.act_upon(result) logging.debug("\x1b[7;34m> set: {} \x1b[0m".format(self.get_value(result))) self.examples.append((decomposition, inferred_tree)) + self.originals.append(example['text']) # Reduce values self.trained = parsing.reprocess_language_knowledge(self, self.examples)