我利用泰坦尼克数据集进行一些神经网络练习。我将数据集拆分为训练和测试。我将火车分成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
值为1
和2
。我如何强制predict
与变量0
中的1
和Survived
匹配,以便表格可以在视觉上匹配?