dtree_sklearn.py 文件源码

def loan_demo():
    dt = tree.DecisionTreeClassifier()
    X, Y = get_loan_data_lh()
    dt = dt.fit(X, Y)
    dot_file = 'loan.dot'
    tree.export_graphviz(dt, out_file=dot_file)
    feature_names = ['age', 'has work', 'own house', 'loan level']
    visualize_tree(dt, feature_names, dot_file)