代码:
svmdata<-read.csv("E://12yue//12quan.csv",header=T)
ind<-sample(2,nrow(svmdata),replace=TRUE,prob=c(0.9,0.1))
traindata<-svmdata[ind==1, ]
testdata<-svmdata[ind==2, ]
tdata <- as.matrix(traindata)
edata <- as.matrix(testdata)
y <- as.matrix(traindata$id)
library(xgboost)
xgb <- xgboost(data=tdata,label=y,max.depth=6,eta=0.3,nrounds=5000,verbose=2,objective="binary:logistic")
pred <- predict(xgb,sdata)