如何可视化决策树?

时间:2014-07-08 07:25:26

标签: machine-learning treeview classification decision-tree

我正在对几组受试者生成的数据进行多类分类。

我有一个包含61个属性和4个组的数据集。我尝试使用以下代码绘制决策树。

>dataset

>fit <- rpart(Group ~ a1 + a2 + a3 + a4 +..... + a61, data=dataset)
>plot(fit)

>text(fit, use.n = TRUE)

其中a1,a2和a3 ..... a61是数据集的属性。

绘制的树非常庞大,不适合桌面..任何人都可以帮助可视化树吗?或者有没有其他方法来修剪树?

1 个答案:

答案 0 :(得分:0)

您可以使用rpart.plot包。 查看this blog post

但是,您应该考虑pruning the tree:您将更容易理解(和绘制)模型。