使用sklearn对机器学习第三篇　决策树进行分析

#-*- coding: utf-8 -*-
from sklearn.tree import DecisionTreeClassifier as DTC
from sklearn.tree import export_graphviz
x = [

[1,1],[1,0],[0,]

y = [ 1,1,0]
dtc = DTC(criterion='entropy')

dtc.fit(x,y)
with open('af.dot','w') as f:

f = export_graphviz(dtc,feature_names=['no surfacing','flippers'],out_file = f)

运行dot命令将dot文件转换成pdf
dot -Tpdf af.dot -o af.pdf

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!