【发布时间】:2017-09-19 14:59:48
【问题描述】:
我使用 sklearn 创建了一个决策树。
from sklearn import tree
clf = tree.DecisionTreeClassifier(max_depth=3)
clf = clf.fit(X, Y)
数据框 X 中的参数是 - 'Company size'、'Industry_other'、'Account size'、'Country' 和 'Use case 1'。
尝试使用export_graphviz 可视化树时,我得到了 2 个相似的节点:
得到相似节点的原因是什么?如何阅读这棵树?
【问题讨论】:
-
你有这个节点在同一层,所以一切正常。这仅意味着,对于大于和小于 0.5 的
Industry_Other,Company size(<= 1.5) 的决策规则是相同的。
标签: python-3.x machine-learning scikit-learn decision-tree supervised-learning