r - 预测是使用不同的类值 Rstudio

Question

我正在利用 Titanic 数据集来执行一些神经网络练习。我将数据集分成训练和测试。我将火车分成 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没有幸存的变量。现在我想用这段代码打印我的混淆矩阵......

table(pred, train.20$Survived)

结果如下......

pred  0  1
   1 70 14
   2 17 42

如果我安慰出pred值是1and 2。我如何强制predict匹配0和1变量一样，Survived以便表格可以在视觉上匹配？