def test_stringify_tree(self):
tree = Node("Gender",
("Female", Node("Bush Approval",
("Approve", Label("McCain")),
("Disapprove", Label("Obama")))),
("Male", Node("Ideology",
("Liberal", Label("Obama")),
("Moderate", Label("Obama")),
("Conservative", Label("McCain")))))
xml_tree = model.stringify_tree(tree)
self.assertEqual(tree, model.build_tree(xml_tree))
def main(domainxml, trainingsetcsv, restrictionstxt):
restrictions = dataset.restrictions_from_text(restrictionstxt)
cols, data = dataset.read(trainingsetcsv.read(), restrictions)
# call train function with:
# `col_sets` - list of sets per column, NOT including class label
# `data` (list of ([train data], class))
tree = Node("swole", ("true", Label("protein and starches")),
("false", Label("sugar"))) # dummy temp tree
tree = c45.run(data, list(enumerate(cols)), 0)
tree_xml = stringify_tree(tree)
sys.stdout.buffer.write(tree_xml)
