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