如何获得单个类别的随机森林的预测概率

时间:2019-02-18 16:49:35

标签: r random-forest

我正在尝试在R版本3中对随机森林建模。我的因变量具有2类,0和1。我想预测1的概率。

所以我尝试了以下代码

cancer.BM.train.rf = randomForest(as.factor(cancer) ~ ., data=cancer.train,ntree=1000, importance=TRUE)
pred=predict(cancer.BM.train.rf, newdata = cancer.test,type="prob")
cancer.test$predProbability=pred

在csv中保存结果的同时,我在csv中获得了两个字段predProbability.0和predProbability.1。但是在Cancer.test数据框中,我找不到predProbability.0。我想将其从我的产品中删除数据。请提出。

0 个答案:

没有答案