回归树

利用R建立回归树模型

参考文献:《数据挖掘与R语言》

1. 安装程序包并载入

2.建模

3.画出决策树

程序包需要载入rpart和DMwR。rpart建立决策树模型，DMwR包有学习集algae。

install.packages('rpart')

install.packages('DMwR')

library(rpart)

library(DMwR)

建模过程：

iris.rp <- rpart(Species~.,data=iris)

algae <- algae[-manyNAs(algae),]    #消除缺失值较多的行

algae.rp <- rpart(a1~.,data = algae[,1:12])

绘制回归树:

直接一个prettyTree搞定

par(mfrow=c(1,2))

prettyTree(iris.rp)

prettyTree(algae.rp)

par(mfrow=c(1,1))

得到如下回归树:

algae:

iris: