def test(): labels = ['no surfacing', 'flippers'] print create_tree([ [1, 1, 'yes'], [1, 1, 'yes'], [1, 0, 'no'], [0, 1, 'no'], [0, 1, 'no'], ], labels[:]) print create_tree([ [1, 1, 'maybe'], [1, 1, 'yes'], [1, 0, 'no'], [0, 1, 'no'], [0, 1, 'no'], ], labels[:])
def test(): tree = create_tree([ [1, 1, 'yes'], [1, 1, 'yes'], [1, 0, 'no'], [0, 1, 'no'], [0, 1, 'no'], ], ['no surfacing', 'flippers']) create_plot(tree, name='test_tree')
def test(): labels = ['no surfacing', 'flippers'] tree = create_tree([ [1, 1, 'yes'], [1, 1, 'yes'], [1, 0, 'no'], [0, 1, 'no'], [0, 1, 'no'], ], labels[:]) print classify(tree, labels, [1, 0]) print classify(tree, labels, [1, 1])
def test(): labels = ['no surfacing', 'flippers'] tree = create_tree([ [1, 1, 'yes'], [1, 1, 'yes'], [1, 0, 'no'], [0, 1, 'no'], [0, 1, 'no'], ], labels[:]) print 'Dump tree:\t', tree dump_tree(tree, 'decision_tree/dump_tree.txt') print 'Loaded tree:\t', load_tree('decision_tree/dump_tree.txt')