我使用随机森林建模。我的数据集目标是预测客户是否会接受特定产品。
因此,我的输出变量有“Yes”、“No”和“”的因子。
我的问题是,我是否有可能在随机森林中预测“是”或“否”?
示例数据如下所示
Outputvar <- c("Yes", "Yes", "No", "NO", "", "")
Inputvar1 <- c("M", "F", "F", "M", "F", "M")
Inputvar2 <- c("34","25","40","50","60","34")
data <- data.frame(cbind(Outputvar,Inputvar2,Inputvar1))
我是R新手,如果我的理解是错误的,那么有人能解释一下我能做些什么吗?
编辑:这是我一直尝试的代码
library(RandomForest)
data$outvar <- factor(data$outputvar, exclude = NULL)
ind0 <- sample(2, nrow(data), replace = TRUE, prob = c(0.7,0.3))
train0 <- data[ind0==1, ]
test0 <- data[ind0==2, ]
fit1 <- randomForest(outputvar1~., data=train0)
print(fit1)
plot(fit1)
编辑2:
编号:3536
是:1061