Python create_treeの例

コード例 #1

ファイル: create_tree.py プロジェクト: beslave/learning

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[:])

コード例 #2

ファイル: draw_tree.py プロジェクト: beslave/learning

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

コード例 #3

ファイル: classify.py プロジェクト: beslave/learning

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

コード例 #4

ファイル: load_dump.py プロジェクト: beslave/learning

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