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

Question

在RI中,可以使用API​​直接绘制对应于CART模型的决策树的图形表示.例如,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应该显示类似

这是笔记本。