预测是使用不同的类值Rstudio

时间:2016-10-22 23:18:06

标签: r predict nnet

我利用泰坦尼克数据集进行一些神经网络练习。我将数据集拆分为训练和测试。我将火车分成80%和20%进行训练。清洁完一切后,我开始训练。这是我的训练代码。

nfit <- train(Survived ~ PassengerId + Pclass + Name + Sex + Age + SibSp +
          Parch + Ticket + Fare + Embarked, 
          data = train.80, method = 'nnet', tuneGrid = grid, 
          skip = TRUE, tuneLength = 2, times = 4)

之后我用这段代码预测了......

pred <- as.numeric(predict(nfit, train.20.survived))

train.20.survived DOESN&T; T上有幸存的变量。现在我想用这段代码打印我的混淆矩阵......

table(pred, train.20$Survived)

结果如下......

pred  0  1
   1 70 14
   2 17 42

如果我控制pred值为12。我如何强制predict与变量0中的1Survived匹配,以便表格可以在视觉上匹配?

0 个答案:

没有答案