install.packages(“randomForest”)
library(randomForest)
d0_train=d0[label0<=5,]
d0_test=d0[label0>5,]
d1_train=d1[label1<=5,]
d1_test=d1[label1>5,]
d2_train=d2[label2<=5,]
d2_test=d2[label2>5,]
d3_train=d3[label3<=5,]
d3_test=d3[label3>5,]
d_train=rbind(d0_train,d1_train,d2_train,d3_train)
d_test=rbind(d0_test,d1_test,d2_test,d3_test)
d_train $ y[d_train$y>=1]=1
d_test $ y[d_test$y>=1]=1
d_train $ y= as.factor(d_train$y)
#令Y变为分类数据(改变数据类型) 不再是int
re_rf = randomForest(y~.-id,data=d_train,ntree=50) #建模,森林默认500 棵树
pred_rf=predict(re_rf,newdata=d_test,type=“prob”) #预测
#画ROC曲线
install.packages(“ROCR”)
library(ROCR)
pred<-prediction(pred_rf[,2],d_test$y)