我怎样才能在Python中绘制CART树,就像在R中一样？

发布时间：2020-12-16 23:53:49 所属栏目：Python 来源：网络整理

导读：在R中,我可以直接使用API??绘制对应于CART模型的决策树的图形表示.例如 prp 会产生类似的东西但我找不到任何类似的API用于Python中的等效功能.例如,尽管我可以告诉sklearn的RandomForestClassifier和 DecisionTreeClassifier 都没有方法或绘图树. 如何在Pyt

在R中,我可以直接使用API??绘制对应于CART模型的决策树的图形表示.例如prp会产生类似的东西

但我找不到任何类似的API用于Python中的等效功能.例如,尽管我可以告诉sklearn的RandomForestClassifier和DecisionTreeClassifier都没有方法或绘图树.

如何在Python中获得CART或随机林树的图形表示？

最佳答案

使用export_graphviz功能.

from sklearn.tree import DecisionTreeClassifier,export_graphviz
np.random.seed(0)
X = np.random.randn(10,4)
y = array(["foo","bar","baz"])[np.random.randint(0,3,10)]
clf = DecisionTreeClassifier(random_state=42).fit(X,y)
export_graphviz(clf)

现在dotty tree.dot应该显示类似的内容

tree visualization

这是notebook.

（编辑：李大同）

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