tree2code.py

def get_code(tree, feature_names, class_names):
    left      = tree.tree_.children_left
    right     = tree.tree_.children_right
    threshold = tree.tree_.threshold
    features  = [feature_names[i] for i in tree.tree_.feature]
    value = tree.tree_.value

    def recurse(left, right, threshold, features, node, indent):
        if threshold[node] != -2:
            print "%sif %s <= %s:" % (indent, features[node], threshold[node])
            if left[node] != -1:
                recurse(left, right, threshold, features, left[node], indent+'    ')
            print '%selse:' % indent
            if right[node] != -1:
                recurse(left, right, threshold, features, right[node], indent+'    ')
        else:
            cls = class_names[np.argmax(value[node])]
            print '%sreturn %s' % (indent, cls)

    recurse(left, right, threshold, features, 0, '')
        
get_code(clf, features, ['False', 'True'])