Question

好的，所以另一个与泰坦尼克号竞赛相关的新手问题：

我正在尝试针对我的测试数据运行随机森林预测。我的所有工作都是在综合测试和培训数据上完成的。

我现在将2拆分为testdata和trainingdata

我有以下代码：

trainingdata <- droplevels(data.combined[1:891,])
testdata <- droplevels(data.combined[892:1309,])

fitRF <- randomForest(as.factor(Survived) ~ Pclass + Sex + Age + SibSp 
+ Parch + Fare + Embarked
                   + new.title + family.size + FamilyID2,
                  data=trainingdata,
                  importance =T,
                  ntree=2000)

varImpPlot(fitRF)

#All works up to this point


Prediction <- predict(fitRF, testdata)
#This line above generates error
submit <- data.frame(PassengerID = data.combined$PassengerId, Survived 
= Prediction)
write.csv(submit, file="14072017_1_RF", row.names = F)

当我运行预测行时，我收到以下错误：

> Prediction <- predict(fitRF, testdata)
Error in predict.randomForest(fitRF, testdata) : 
  New factor levels not present in the training data

我跑的时候 str（testdata）和str（trainingdata）我可以看到2个不再匹配的因素

Trainingdata      
$ Parch            : Factor w/ 7 levels 

Testdata
$ Parch            : Factor w/ 8

Trainingdata
$ FamilyID2        : Factor w/ 22 

Testdata
$ FamilyID2        : Factor w/ 18

这些差异导致我的错误发生了吗？如果是这样，我该如何解决这个问题？

非常感谢

其他信息：我已从RandomForest创建行中删除了Parch和FamilyID2，现在代码正常工作，因此肯定是那两个导致问题的级别不匹配的变量。

Answer 1

这里的新手，我这几天只是在玩泰坦尼克号。我认为将Parch变量作为一个因素是没有意义的，所以也许把它变成数字并且可以解决问题：

训练$ Parch＆lt; - as.numeric（train $ Parch）

否则，测试数据有2个遮挡，其值为9，对于Parch，列车数据中不存在：

> table(train$Parch)

0   1   2   3   4   5   6 
678 118  80   5   4   5   1 

> table(test$Parch)

0   1   2   3   4   5   6   9 
324  52  33   3   2   1   1   2 
>

或者，如果您需要将变量作为一个因子，那么您可以只为其添加另一个级别：

train$Parch <- as.factor(train$Parch) # in my data, Parch is type int
train$Parch
levels(train$Parch) <- c(levels(train$Parch), "9") 
train$Parch # now Parch has 7 levels
table(train$Parch) # level 9 is empty

R中的随机森林：训练数据中不存在的新因子水平

1 个答案: