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