从零开始学Python【27】--Logistic回归(实战部分)

浏览: 6002

往期精彩回顾

从零开始学Python【26】--Logistic回归(理论部分)

从零开始学Python【25】--岭回归及LASSO回归(实战部分)

从零开始学Python【24】--岭回归及LASSO回归(理论部分)

从零开始学Python【23】--线性回归诊断(第二部分)

从零开始学Python【22】--线性回归诊断(第一部分)

从零开始学Python【21】--线性回归(实战部分)

从零开始学Python【20】--线性回归(理论部分)

前言


      基于上一期的理论知识,我们本期跟大家分享一下如何通过Python和R语言完成Logistic回归分类器的构建。大家都知道,Logistic模型主要是用来解决二元分类问题,通过构建分类器,计算每一个样本为目标分类的概率,一般而言,我们会将概率值0.5作为分类的阈值,即概率值P大于等于0.5时判别为目标分类,否则为另一种分类。

      本次分享的数据是基于用户信息(年龄、性别和年收入)来判断其是否发生购买,数据来源于GitHub(文末有数据链接可供下载)。接下来,让我们看看Logistic模型是如何完成二分类问题的落地。本次分享会涉及模型的构建、测试集的预测及模型的验证三个方面。

image.png

数据查阅


# ===== Python3 =====

# 导入第三方包
import pandas as pd
import numpy as np
import statsmodels.formula.api as smf
from sklearn.cross_validation import train_test_split
from sklearn import metrics
from ggplot import *

# 读取数据集
purchase = pd.read_csv(r'E:\Jupyter\ML\Logistic-Regression-master\Social_Network_Ads.csv')
# 查看数据类型
purchase.dtypes
# 查看各变量的缺失情况
purchase.isnull().sum()

image.png

从上图结果可知,除Gender变量,其余变量均为数值型变量,那么待会再构建入模变量时,需要对Gender变量创建哑变量;一般在做数据探索时,需要检查各变量是否存在缺失的情况(如果缺失需要借助于删除法、替换法、插值法等完成缺失值的处理,具体可以参考文章【如何使用R语言解决可恶的脏数据】),很显然上面的结果并没有显示数据中含有缺失值。

变量处理

# 对Gender变量作哑变量处理
dummy = pd.get_dummies(purchase.Gender)
# 为防止多重共线性,将哑变量中的Female删除
dummy_drop = dummy.drop('Female', axis = 1)

# 剔除用户ID和Gender变量
purchase = purchase.drop(['User ID','Gender'], axis = 1)
# 如果调用Logit类,需要给原数据集添加截距项
purchase['Intercept'] = 1

# 哑变量和原数据集合并
model_data = pd.concat([dummy_drop,purchase], axis = 1)
model_data

image.png

上面所做的处理无非是构建哑变量,然后从哑变量中再剔除一个水平变量(这个非常重要,为了防止多重共线性);同时还要剔除没有意义的变量User ID和不再使用的Gender变量(因为已经拆分为哑变量了)。OK,整理之后的数据集就如上图所示,接下来我们要基于这个数据集进行Logistic模型的创建。

Logistic模型

# 将数据集拆分为训练集和测试集
X = model_data.drop('Purchased', axis = 1)
y = model_data['Purchased']
# 训练集与测试集的比例为75%和25%
X_train, X_test, y_train, y_test = train_test_split(X, y, train_size = 0.75, random_state=0)

# 根据训练集构建Logistic分类器
logistic = smf.Logit(y_train,X_train).fit()
logistic.summary()

image.png

经过7次迭代后,模型系数的计算实现收敛,完成了Logistic分类器的创建。从上图的结果看,除Male这个哑变量不显著(说明性别这个变量并不能构成用户是否购买的因素)外,其余的偏回归系数均为显著。下面,我们不妨将Male变量剔除,再做一次Logistic模型。

# 重新构造分类器
logistic2 = smf.logit('Purchased~Age+EstimatedSalary',
                     data = pd.concat([y_train,X_train], axis = 1)).fit()
logistic2.summary()

print(logistic.aic)
print(logistic2.aic)

image.png

image.png

很明显,通过变量的剔除,在保证了所有变量显著的情况下,也降低了模型的AIC,说明,Male哑变量的删除是合理的。

# 优势比
np.exp(logistic2.params)

image.png

接下来,再来看看模型的系数解释(优势比),在其他变量不变的情况下,用户年龄每增加一个单位(岁),用户购买的概率是不够买概率的1.25倍;用户的年收入每增加一个单位(元),用户购买的概率与不够买概率几乎相等(因为这里只是计算年收入增加1元的概率比,如果对收入变量压缩10000倍,那这个概率比肯定就会上升了,因为此时收入上升一个单位就是一万元了)。

模型预测与验证

# 根据分类器,在测试集上预测概率
prob = logistic2.predict(exog = X_test.drop('Male', axis = 1))
# 根据概率值,将观测进行分类,不妨以0.5作为阈值
pred = np.where(prob >= 0.5, 1, 0)

# 根据预测值和实际值构建混淆矩阵
cm = metrics.confusion_matrix(y_test, pred, labels=[0,1])
cm
# 计算模型的准确率
accuracy = cm.diagonal().sum()/cm.sum()
accuracy

image.png

应用分类器对测试数据集进行预测,这里将概率值设为0.5,如果概率大于等于0.5则判用户会购买,否则不会发生购买。通过这个概率值的设定,我们发现模型的准确率还是非常高的(混淆矩阵对角线代表预测正确的数量)。可是单看混淆矩阵还不够,因为当数据不平衡时,计算的准确率也同样会高,并不代表模型就会好,所以我们进一步的借助于ROC曲线下的面积来衡量模型时候合理。

# 绘制ROC曲线
fpr, tpr, _ = metrics.roc_curve(y_test, pred)
df = pd.DataFrame(dict(fpr=fpr, tpr=tpr))

ggplot(df, aes(x='fpr', y='tpr')) +\
   geom_area(alpha=0.5, fill = 'steelblue') +\
   geom_line() +\
   geom_abline(linetype='dashed',color = 'red') +\
   labs(x = '1-specificity', y = 'Sensitivity',title = 'ROC Curve AUC=%.3f' % metrics.auc(f

image.png

是不是很激动,对于熟悉R语言的你,Python中也有ggplot2的绘图语法!从上面的ROC曲线结果可知,AUC的值超过了0.85,这进一步说明模型的预测效果是非常不错的(一般AUC>0.8就比较好了)。

      到此,关于使用Python构建Logistic分类器的实战我们就介绍到这里,接下来将使用R语言重新复现一遍,希望对R语言熟悉的朋友有一点的帮助。如下是R语言的复现脚本:

# 加载第三方包
library
(pROC)
library(ggplot2)

# 读取数据
purchase <- read.csv(file = file.choose())
purchase$Purchased = factor(purchase$Purchased)

# 数据类型
str(purchase)
# 各变量缺失情况
sapply(purchase, function(x) sum(is.na(x)))

# 数据集拆分为训练集和测试集
set.seed(0)
idx = sample(1:nrow(purchase), size = 0.75*nrow(purchase))
train = purchase[idx,]
test = purchase[-idx,]

# 构建Logistic模型
logit = glm(Purchased ~ ., data = train[,-1], family = binomial(link = "logit"))
summary(logit)

# 删除性别这个变量
train2 = purchase[idx,-c(1,2)]
test2 = purchase[-idx,-c(1,2)]

# 重新构建Logistic模型
logit2 = glm(Purchased ~ ., data = train2, family = binomial(link = "logit"))
summary(logit2)

# 预测
prob = predict(logit2, newdata = test2)
pred = factor(ifelse(prob >= 0.5, 1, 0), levels = c(0, 1), ordered = TRUE)

# 混淆矩阵
Freq = table(test$Purchased, pred)
# 准确率
sum(diag(Freq))/sum(Freq)

# 绘制ROC曲线
ROC = roc(factor(test2$Purchased, levels=c(0,1), ordered = TRUE), pred)
df = data.frame(x = 1-ROC$specificities, y = ROC$sensitivities)

ggplot(df, aes(x = x, y = y)) +
 geom_area(alpha=0.5, fill = 'steelblue') +
 geom_line() +
 geom_abline(linetype='dashed',color = 'red') +
 labs(x = '1-specificity', y = 'Sensitivity',
      title = paste0('ROC Curve AUC=', round(ROC$auc,3))) +
 theme(plot.title = element_text(hjust = 0.5, face = 'bold'))

结语


      OK,关于使用Python和R语言完成Logistic回归的实战我们就分享到这里,如果你有任何问题,欢迎在公众号的留言区域表达你的疑问。同时,也欢迎各位朋友继续转发与分享文中的内容,让更多的人学习和进步。

关注“每天进步一点点2015”,与小编同进步!

数据链接:

链接: https://pan.baidu.com/s/1eSo3Y2Q 密码: cd7f

推荐 0
本文由 每天进步一点点2015 创作,采用 知识共享署名-相同方式共享 3.0 中国大陆许可协议 进行许可。
转载、引用前需联系作者,并署名作者且注明文章出处。
本站文章版权归原作者及原出处所有 。内容为作者个人观点, 并不代表本站赞同其观点和对其真实性负责。本站是一个个人学习交流的平台,并不用于任何商业目的,如果有任何问题,请及时联系我们,我们将根据著作权人的要求,立即更正或者删除有关内容。本站拥有对此声明的最终解释权。

0 个评论

要回复文章请先登录注册