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

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

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

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

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

最佳答案使用export_graphviz功能.

from sklearn.tree import DecisionTreeClassifIEr,export_graphviznp.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应该显示类似的内容

这是notebook. 总结

以上是内存溢出为你收集整理的我怎样才能在Python中绘制CART树,就像在R中一样？全部内容，希望文章能够帮你解决我怎样才能在Python中绘制CART树,就像在R中一样？所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。