在 R 中,我可以绘制对应于 CART 的决策树的图形表示。直接使用 API 建模。例如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
应该显示类似的东西
这是一个 notebook .
关于python - 如何像在 R 中那样在 Python 中绘制 CART 树?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/22824452/