python 实现决策树规则

python 实现决策树规则

----决策树探索----
from sklearn import tree
import graphviz
import pydotplus
from pydotplus.graphviz import graph_from_dot_data
from sklearn.tree import export_graphviz
import os   
os.environ["PATH"] += os.pathsep + 'D:/软件安装/windows_10_msbuild_Release_graphviz-9.0.0-win32/Graphviz/bin/'

x = bj_test_union[['x1','x2','x3','x4']]
y = bj_test_union["dpd30"]
cart_tree = tree.DecisionTreeRegressor(max_depth=2)
cart_tree = cart_tree.fit(x,y)


cart_tree_data = tree.export_graphviz(
cart_tree,
out_file =None,
proportion=True,
feature_names=['x1','x2','x3','x4'],
class_names=["0","1"],
filled=True,
rounded=True,
special_characters=True)
graph = graphviz.Source(cart_tree_data)
graph