python - 如何像在 R 中那样在 Python 中绘制 CART 树？

Question

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

但是我找不到任何类似的 API 来实现 Python 中的等效功能。例如，据我所知，sklearn 的 RandomForestClassifier 和 DecisionTreeClassifier 都不是。有方法或画树。

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

Answer 1

使用 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 .