继续浏览精彩内容
慕课网APP
程序员的梦工厂
打开
继续
感谢您的支持,我会继续努力的
赞赏金额会直接到老师账户
将二维码发送给自己后长按识别
微信支付
支付宝支付

使用R完成逻辑斯蒂回归分类 直接上代码,如下:

慕斯卡3302699
关注TA
已关注
手记 271
粉丝 58
获赞 314

data_sample <- iris[51:150,];
m <- dim(data_sample)[1]  #获取数据集记录条数
val <- sample(m, size =round(m/3), replace = FALSE, prob= rep(1/m, m))  #抽样,选取三分之二的数据作为训练集。 
iris.learn <- data_sample[-val,]  #选取训练集 
iris.valid <- data_sample[val,]   #选取验证集

#调用glm函数训练逻辑斯蒂二元模型
#glm()提供正态、指数、gamma、逆高斯、Poisson、二项分布。我们的logistic回归使用的是二项分布族binomial。Binomial族默认连接函数为logit,可设置为probit。
logit.fit <- glm(Species~Petal.Width+Petal.Length,
 family = binomial(link = 'logit'),
 data = iris.learn);

#生成测试数据集,实际上直接使用iris.valid
dfrm <- data.frame(Petal.Width=iris.valid$Petal.Width,
 Petal.Length=iris.valid$Petal.Length);

real_sort <- iris.valid$Species;  #测试数据集实际类别

prdict_res <- predict(logit.fit, type="response", newdata=dfrm);  #预测数据产生概率
data.frame(predict=prdict_res, real=real_sort);  #查看数据产生概率和实际分类的关系
data.frame(predict=ifelse(prdict_res>0.5, "virginica", "versicolor"), real=real_sort);  #根据数据产生概率生成预测分类
table(data.frame(predict=ifelse(prdict_res>0.5, "virginica", "versicolor"), real=real_sort));  #计算分类准确度



使用R完成逻辑斯蒂回归分类 直接上代码,如下: - 黄大仙 - 黄大仙


打开App,阅读手记
0人推荐
发表评论
随时随地看视频慕课网APP