在R的隐藏层中选择数量的神经元

时间:2020-04-28 13:07:46

标签: r

嗨,我是R的新手,我正在寻找auc在改变神经元隐藏层的数量,但是代码不好


    # Generate progress bar
pbar <- create_progress_bar('text')
pbar$init(3)

set.seed(100)
for(i in 1:3)
{
  # Fit the net and calculate training error (point estimate)
  nmodel <- neuralnet(f,data=training,hidden=c(i),linear.output=T)
  #prediction 
  output <- neuralnet::compute(nmodel, testing[,-c(1,2)],rep=1)

  prob.result<- output$net.result
  results <- data.frame(Actual=testing[,"P1102inacPcast"],Prediction=output$net.result)
  #head(as.matrix(testing[,"P1102inacPcast"]))
  Actual= testing[,"P1102inacPcast"]
  #head (as.data.frame(Actual))
  nn.pred =ROCR::prediction(prob.result, as.matrix(Actual))

  #roc curve
  roc <- performance(nn.pred, "tpr", "fpr")
  plot(roc)
  abline(a=0,b=1)
# auc calcul
  auc<- performance(nn.pred, "auc")
  auc<-unlist(slot(auc,"y.values"))
  auc

0 个答案:

没有答案