Save original examples (needed on show_samples).

This commit is contained in:
kenkeiras 2017-09-30 00:55:42 +02:00
parent d607b2210e
commit 2c36dd9b7e

View File

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