cda

您的位置:首页 > 精彩阅读 > r语言做决策树代码实现

r语言做决策树代码实现

2018-06-14

r语言做决策树代码实现

0.节点和结点的区别:节点为两线相交,不为终点;而结点为两线相交为终点,没有延伸;
1.分支节点:它指向其他的节点,所以是度不为0的节点。 vs 叶子结点:度为0的结点
2.度:结点拥有的子树数;就是说这个结点下面有几条分支
3.树的深度:树有几层
4.10折交叉验证:常用的测试算法准确性的方法。
  将数据集分成10份,轮流将其中9份作为训练数据,1份作为测试数据,进行试验
  每次试验都会得出相应的正确率,10次结果的正确率取平均值就作为算法精度的估计,一般还需进行多次10折交叉验证,再求均值
  为什么取10折?因为很多理论证明了10折是获得最好误差估计的恰当选择。
  
#第1步:工作目录和数据集的准备
setwd("C:/Users/IBM/Desktop/170222分类树建模/2.23建模")#设定当前的工作目录,重要!
audit2<-read.csv("model2.csv",header=T)
str(audit2) #转成字符串类型的

#第2步:做训练集和测试集
set.seed(1)
sub<-sample(1:nrow(audit2),round(nrow(audit2)*2/3))
length(sub)
data_train<-audit2[sub,]#取2/3的数据做训练集
data_test<-audit2[-sub,]#取1/3的数据做测试集
dim(data_train)#训练集行数和列数13542 23
dim(data_test) #测试集的行数和列数6771 23
table(data_train$是否转化) #看该列分布的
table(data_test$是否转化)

#做决策树模型。首先对树参数进行设置,再建模
## rpart.control对树进行一些设置  
## xval是10折交叉验证  
## minsplit是最小分支节点数,这里指大于等于20,那么该节点会继续分划下去,否则停止  
## minbucket:叶子节点最小样本数,这里设置100,可以调参 
## maxdepth:树的深度  

## cp全称为complexity parameter,指某个点的复杂度,对每一步拆分,模型的拟合优度必须提高的程度 

#加载程序包和一些参数设定
library(rpart)
ct<-rpart.control(xval=10,minsplit=20,minbucket=150,cp=0.00017)

#rapart包中的raprt函数来做决策树
#na.action:缺失数据的处理,默认为删因变量缺失保留自变量缺失
#method:树的末端数据类型选择相应的变量分割方法:
#        连续性method=“anova”,离散型method=“class”,计数型method=“poisson”,生存分析型method=“exp”  
#parms:用来设置三个参数:先验概率、损失矩阵、分类纯度的度量方法(gini和information)  

#第3步:建模,观察模型结果

library(rpart)

tree.both<-rpart(as.factor(是否转化)~ .,data=data_train,method='class',minsplit=20,minbucket=150,cp=0.00017)
summary(tree.both)
tree.both$variable.importance
printcp(tree.both)
plotcp(tree.both,lwd=2) 

#第4步:画决策树
#画决策树第1种方法,画出来的树比较简单
par(mfrow=c(1,3))
plot(tree.both)
text(tree.both,use.n=T,all=T,cex=0.9)

#画决策树第2种方法,画出来的树稍微好看些
library(rpart.plot)
rpart.plot(tree.both,branch=1,shadow.col="gray",box.col="green",border.col="blue",split.col="red",split.cex=1.2,main="决策树")

#第5步:剪枝
#rpart包提供了一种剪枝方法--复杂度损失修剪的修剪方法
#printcp这个函数会告诉你分裂到的每一层,对应的cp是多少,平均相对误差是多少
#xerror:交叉验证的估计误差;xstd:标准误差;xerror±xstd平均相对误差
printcp(tree.both)

#我们使用具有最小交叉验证误差的cp
cp=tree.both$cptable[which.min(tree.both$cptable[,"xerror"]),"CP"]

cp #cp=0.00049


#第6步:剪枝之后的树再画图
tree.both2<-prune(tree.both,cp=tree.both$cptable[which.min(tree.both$cptable[,"xerror"]),"CP"])
summary(tree.both2)
tree.both2$variable.importance
printcp(tree.both2)
plotcp(tree.both2,lwd=2) 

library(rpart.plot)
rpart.plot(tree.both2,branch=1,shadow.col="gray",box.col="green",border.col="blue",split.col="red",split.cex=1.2,main="决策树")

#第7步:输出规则。剪枝后的决策树规则,从规则中再发现规律
library(rattle)
asRules(tree.both2)

#第8步:在测试集上做预测
library(pROC)

pred.tree.both<-predict(tree.both,newdata=data_test)

#第9步,看测试的效果,预测正确的有多少,预测错误的有多少

predictScore<-data.frame(pred.tree.both)
rownames(predictScore) #看这个矩阵行的名字
colnames(predictScore)#看这个矩阵列的名字
predictScore$是否转化<-'ok'  #在预测的矩阵后面多加一列‘是否转化’2,全部都是2
predictScore[predictScore$FALSE.>predictScore$TRUE.,][,"是否转化"]=FALSE #如果false的概率大于true的概率,那么判断为false
predictScore[predictScore$FALSE.<=predictScore$TRUE.,][,"是否转化"]=TRUE 
n<-table(data_test$是否转化,predictScore$是否转化)
n #看分布情况
percantage<-c(n[1,1]/sum(n[1,]),n[2,2]/sum(n[2,]))
percantage

完 谢谢观看

分享
收藏

OK