diff --git a/naive-nlu/tree_nlu/cli.py b/naive-nlu/tree_nlu/cli.py index 1306d83..bd4e894 100644 --- a/naive-nlu/tree_nlu/cli.py +++ b/naive-nlu/tree_nlu/cli.py @@ -1,3 +1,4 @@ +import logging from .knowledge_base import KnowledgeBase from .tests import gac_100 from .modifiable_property import ( @@ -16,7 +17,9 @@ def debug(knowledge): def main(): + logging.getLogger().setLevel(logging.INFO) knowledge = gac_100.main() + logging.getLogger().setLevel(logging.DEBUG) while True: try: data = input("> ").strip() diff --git a/naive-nlu/tree_nlu/knowledge_base.py b/naive-nlu/tree_nlu/knowledge_base.py index 33bd9e9..6386ce4 100644 --- a/naive-nlu/tree_nlu/knowledge_base.py +++ b/naive-nlu/tree_nlu/knowledge_base.py @@ -30,18 +30,18 @@ class KnowledgeBase(object): }) self.act_upon(result) - logging.info("\x1b[7;32m> {} \x1b[0m".format(example)) + logging.debug("\x1b[7;32m> {} \x1b[0m".format(example)) tokens, decomposition, inferred_tree = parsing.integrate_language(self, example) - logging.info(tokens) + logging.debug(tokens) result = knowledge_evaluation.integrate_information(self.knowledge, { "elements": tokens, "decomposition": decomposition, "parsed": inferred_tree, }) - logging.info("\x1b[7;33m< {} \x1b[0m".format(self.get_value(result))) + logging.debug("\x1b[7;33m< {} \x1b[0m".format(self.get_value(result))) self.act_upon(result) - logging.info("\x1b[7;34m> set: {} \x1b[0m".format(self.get_value(result))) + logging.debug("\x1b[7;34m> set: {} \x1b[0m".format(self.get_value(result))) self.examples.append((decomposition, inferred_tree)) # Reduce values @@ -57,7 +57,7 @@ class KnowledgeBase(object): def process(self, row): row = row.lower() knowledge_before = copy.deepcopy(self.knowledge) - logging.info("\x1b[7;32m> {} \x1b[0m".format(row)) + logging.debug("\x1b[7;32m> {} \x1b[0m".format(row)) tokens = parsing.to_tokens(row) tokens, inferred_tree = parsing.get_fit(self, tokens) result = knowledge_evaluation.integrate_information(self.knowledge, diff --git a/naive-nlu/tree_nlu/tests/basic.py b/naive-nlu/tree_nlu/tests/basic.py index b570608..414a0ce 100644 --- a/naive-nlu/tree_nlu/tests/basic.py +++ b/naive-nlu/tree_nlu/tests/basic.py @@ -110,13 +110,13 @@ base_knowledge = { } def test_assumption(expectedResponse, knowledge, query): - logging.info("Query: {}".format(query['text'])) - logging.info("Expected: {}".format(expectedResponse)) + logging.debug("Query: {}".format(query['text'])) + logging.debug("Expected: {}".format(expectedResponse)) result, abstract_tree, diff = knowledge.process(query['text']) end_result = result.getter() if is_modifiable_property(result) else result - logging.info("\x1b[0;3{}mResult: {}\x1b[0m".format("1" if end_result != expectedResponse else "2", end_result)) + logging.debug("\x1b[0;3{}mResult: {}\x1b[0m".format("1" if end_result != expectedResponse else "2", end_result)) if end_result != expectedResponse: raise AssertionError('{} is not {}'.format(end_result, expectedResponse)) @@ -127,9 +127,9 @@ def main(): differences = knowledge.train(examples) - logging.info("----") - logging.info(differences()) - logging.info("----") + logging.debug("----") + logging.debug(differences()) + logging.debug("----") test_assumption(True, knowledge, {'text': 'earth is a planet'}) test_assumption(True, knowledge, {'text': 'is lava dangerous?'}) @@ -137,12 +137,12 @@ def main(): row = test['text'] result, inferred_tree, differences = knowledge.process(row) - logging.info("result:", result) - logging.info(differences()) - logging.info("---") - logging.info('-----') - logging.info(json.dumps(sorted(knowledge.knowledge.keys()), indent=4)) - logging.info('-----') + logging.debug("result:", result) + logging.debug(differences()) + logging.debug("---") + logging.debug('-----') + logging.debug(json.dumps(sorted(knowledge.knowledge.keys()), indent=4)) + logging.debug('-----') queryTrue = { "text": "is io a moon?", diff --git a/naive-nlu/tree_nlu/tests/gac_100.py b/naive-nlu/tree_nlu/tests/gac_100.py index 484ebd3..34f4c3b 100644 --- a/naive-nlu/tree_nlu/tests/gac_100.py +++ b/naive-nlu/tree_nlu/tests/gac_100.py @@ -701,6 +701,7 @@ def main(): for i, (example_type, data) in enumerate(examples): if example_type == 'full_example': + logging.info(data['affirmation']) affirmation = { 'text': data['affirmation'], 'parsed': data['parsed'][1], @@ -724,6 +725,8 @@ def main(): raise AssertionError('{} is not {}'.format(result, data['answer'])) elif example_type == 'text_example': + logging.info(data['affirmation']) + show_progbar(i, total, data['affirmation']) affirmation = data['affirmation'] logging.debug("Processing affirmation: {}".format(affirmation))