在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)
这是notebook.
原文地址:https://www.jb51.cc/python/439749.html
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。