R语言实现随机森林建模的实例

问题,用户注册七天内,通过用户的浏览行为判断用户是否会付费?

输入变量:
pv(浏览量)、hotdays(活跃天数)、time(活跃时长)、aciton(关键动作数)

目标:
ispay(是否付费)

R程序过程:

# 读取数据
a <- read.delim("~/rwork/data")

# 数据详情如下
str(a)

221003shttxxq5hzh55gtx

 

# 随机森林建模
tree<-randomForest(as.factor(ispay)~.,data=a)

pr<-predict(tree, newdata=a)

table(a$ispay,pr,
dnn=c("Actual", "Predicted"))

# 评估模型
pcme <- function(actual, cl)
{
x <- table(actual, cl)
tbl <- cbind(round(x/length(actual), 2),
Error=round(c(x[1,2]/sum(x[1,]),
x[2,1]/sum(x[2,])), 2))
names(attr(tbl, "dimnames")) <- c("Actual", "Predicted")
return(tbl)
};

pcme(a$ispay, pr)

结果显示,建模正确率高达100%

发表评论

您的电子邮箱地址不会被公开。 必填项已用*标注