【Python学习】决策树生成与可视化

from sklearn.tree import DecisionTreeClassifier as DTC
dtc = DTC(criterion=‘entropy’) #建立决策树模型，基于信息熵
dtc.fit(x, y) #训练模型
#导入相关函数，可视化决策树。
#导出的结果是一个dot文件，需要安装Graphviz才能将它转换为pdf或png等格式。
from sklearn.tree import export_graphviz
x = pd.DataFrame(x)
x = pd.DataFrame(x)
with open(“tree.dot”, ‘w’) as f:
f = export_graphviz(dtc, feature_names = x.columns, out_file = f)