def test_norvig_data(self):
from data_set import build_norvig_test_records
data_set = DataSet(build_norvig_test_records())
root_node = ET.Element('root')
decision_tree(data_set, data_set.attributes, root_node)
with open('norvig_decision_tree.xml', 'wb') as out:
out.write(ET.tostring(prep_e_tree_for_parsing(root_node)))
def test_dummy_tree(self):
dummy_data = [{'age' : 'y', 'class' : True},
{'age' : 'o', 'class' : False}]
dummy_data_set = DataSet(dummy_data)
root_node = ET.Element('root')
decision_tree(dummy_data_set, dummy_data_set.attributes, root_node)
print ET.tostring(prep_e_tree_for_parsing(root_node))
def test_decision_tree(self):
data_set = self.data_set
root_node = ET.Element('root')
decision_tree(data_set, data_set.attributes, root_node)
with open('loan_decision_tree.xml', 'wb') as out:
out.write(ET.tostring(prep_e_tree_for_parsing(root_node)))
