二分类预测出来只有一个值 ,求助
代码如下:
library(rattle)
audit=audit[,-c(1,11,12)]
audit=na.omit(audit)
index=sample(nrow(audit),nrow(audit)*0.75)
train=audit[index,]
test=audit[-index,]
#构造one hot endoding数据
library(caret)
library(magrittr)
dummies <- dummyVars( ~ ., data = train)
trainn <- predict(dummies, newdata = train) %>% as.data.frame()
dummies2 <- dummyVars( ~ ., data = test)
testn <- predict(dummies2, newdata = test) %>% as.data.frame()
#建模
library(neuralnet)
coly=51
formula <- names(trainn[,-coly]) %>% paste(collapse = "+")
formula <- paste("TARGET_Adjusted ", formula, sep = "") %>% as.formula()
nn <- neuralnet(formula = formula, data = trainn,
hidden = c(2), err.fct = "ce",
linear.output = F)
pred=compute(nn,testn[-coly])
result=pred$net.result;result
idx=apply(result,1,which.max)
table(testn$TARGET_Adjusted,idx,dnn=c('真实值','预测值'))