# dt = DecisionTree("tennis")
# # # dt.information_gain("tennis","Wind")
# dt.build_tree("tennis")
# print()
# p = Predictor("tennis-attr.txt")
# p.load_test_data("tennis-test.txt")
# p.all_tests_ruleset(dt.ruleset)
# print("-" * 20)
# p.all_tests_tree(dt.root_node)
dt1 = DecisionTree("tennis")
dt1.build_tree()
p1 = Predictor("tennis-attr.txt")
p1.load_test_data("tennis-test.txt")
p1.all_tests_ruleset(dt1.ruleset)
p1.all_tests_tree(dt1.root_node)

dt2 = DecisionTree("bool")
dt2.build_tree()
p2 = Predictor("bool-attr.txt")
p2.load_test_data("bool-test.txt")
p2.all_tests_ruleset(dt2.ruleset)
p2.all_tests_tree(dt2.root_node)

dt3 = DecisionTree("iris")
dt3.build_tree()
p3 = Predictor("iris-attr.txt")
p3.load_test_data("iris-test.txt")
p3.all_tests_ruleset(dt3.ruleset)
p3.all_tests_tree(dt3.root_node)