R预测因子变量

R predict factor variables

我尝试在 R 中做一些预测。我加载并清理了数据,拟合了一个模型并做了一个看起来不错的预测。我现在的问题是,我的预测给出了某个因素出现的概率百分比,而不是因素本身:

我有一个关于人们如何进行某些锻炼的数据集。这种表现在 A-D 中是不确定的(这是我数据集中的一个因子变量)。当我进行预测时,我得到以下输出:

但我想要这样:

[ B A E A A C D A A A C ]

我该怎么做?这是我的代码:

modFitA1 <- rpart(classe ~ ., data=PML_Train_red, method="class")     
Predictn<-predict(modFitA1, newdata= PML_Test_red)
Predictn

即使您将 method="class" 放入您的模型语句中,您也需要将 type="class" 添加到您的预测语句中。

Predictn<-predict(modFitA1, newdata= PML_Test_red, type="class")