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

标签 python r graph tree scikit-learn

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

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

如何在 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 应该显示类似的东西

tree visualization

这是一个 notebook .

关于python - 如何像在 R 中那样在 Python 中绘制 CART 树?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/22824452/

相关文章:

r - 将空格分隔的文本文件转换为命名向量以计算 HWE

r - 使用 lapply 进行公式改变的多元回归,而不是数据集

graph - 图结构中的拥有指针

algorithm - 如何找到图的加权最小顶点覆盖

python - Django Rest Framework 中带有请求参数的 GET 请求

python - 将 XOR 更改为 XNOR

php - 在 WAMP 环境中设置 Concerto 自适应测试平台时出错

c++ - 第二次在图上运行时,广度/深度优先搜索会导致崩溃

python - DAG(一种连通二叉树)中的所有路径

python - 为什么在 python 中执行追加操作后列表产品会重复元素