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)