练习1-回归模型

近几天详细看了几个模型,很多模型在理解算法思想后其实不难抓住思路,但是实际操作起来会遇到很多问题。

本周就先从回归模型进行巩固并熟悉算法和代码。数据来源于之前数据挖掘群群友分享的网站资源。本文选择的案例集为,预测医疗费用。

主要思路:
1.初步探索数据
2.探索特征之间相关关系
3.处理数据并建模

初步探索数据

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
%matplotlib inline

读取数据

data=pd.read_csv(r'/Users/---/Downloads/医疗费用预测数据集/insurance.csv')

预览数据

data.head()
image.png

画散点图,看特征数据关系

sns.pairplot(data,x_vars=['age','bmi','children'],y_vars='charges',size=7,aspect=0.8,kind='reg')
#用 kind 可以显示预测的回归线和置信区间
image.png

查看相关矩阵的数据

data.describe()
image.png

看到 费用均值>中位数,因此呈现右偏现象;

import pylab as pl
pl.hist(data['charges'])
pl.xlabel('charges')
pl.show()
image.png

在我们的数据中,绝大多数的个人每年的医疗费用都在0~15000美元,尽管分布的尾部经过直方图的峰部后延伸得很远。因为线性回归假设因变量的分布为正态分布,所以这种分布是不理想的。在实际应用中,线性回归的假设往往会被违背。如果需要,我们在后面能够修正该假设。 即将面临的另一个问题就是回归模型需要每一个特征都是数值型的,而在我们的数据中,我们有3个非数值类型的特征。 变量sex被划分成male和female两个水平,而变量smoker被划分成yes和no两个水平。从describe()的输出中,我们知道变量region有4个水平,但我们需要仔细看一看,它们是如何分布的。

#非数值型数据的描述
data['region'].describe() 

count 1338
unique 4
top southeast
freq 364
Name: region, dtype: object

data.region.value_counts()

southeast 364
northwest 325
southwest 325
northeast 324
Name: region, dtype: int64

可以看到,这里的数据几乎平均分在四个地理区域中

data['sex'].describe()

count 1338
unique 2
top male
freq 676
Name: sex, dtype: object

data['smoker'].describe()

count 1338
unique 2
top no
freq 1064
Name: smoker, dtype: object

探索特征之间的关系——相关系数矩阵

在使用回归模型拟合之前,需要确定自变量和因变量之间or 自变量之间是如何相关的。
用相关系数矩阵来快速确定这种相关关系。
命令:corr()

corrmat=data.corr()  #自动对数值型形成数据间对相关系数矩阵
corrmat
image.png
f,ax=plt.subplots(figsize=(20,9))# 控制绘图的尺寸
sns.heatmap(corrmat,vmax=0.9,square=True) # 热力图数据用corrmat,标签最大值为0.9,采用方形 
image.png

从相关系数矩阵结合热力图可以看出,变量之间两两关系如何。 行列之间交叉点位置,就是相关系数(热力图用颜色表示),对角线始终为1,因为变量自身于自身永远正相关(白色表示相关性最强)
从图中看,大多以深色为主,相关系数不是强相关。但是有一些显著关联。比如age和bmi,age和charge,bmi & charges,呈现中度相关。

基于数据训练模型

线性回归模型只能处理数值数据,因此需要对所有数据数值化。首先需要将非数值型变量转化为虚拟变量——sex 变为(0,1)等等
命令:dummycoding()

def dummycoding(dataframe):
#首先删掉数值型变量,对类别变量进行虚拟化
        dataframe_age = dataframe['age']
        dataframe_bmi = dataframe['bmi']
        dataframe_children = dataframe['children']
        dataframe_charges = dataframe['charges']
        dataframe_1 = dataframe.drop(['age'], axis = 1)
        dataframe_2 = dataframe_1.drop(['bmi'], axis = 1)
        dataframe_3 = dataframe_2.drop(['children'], axis = 1)
        dataframe_new=dataframe_3.drop(['charges'],axis=1)
        
        dataframe_new=pd.get_dummies(dataframe_new,prefix=dataframe_new.columns).astype(int)
        
    #对虚拟化完成之后对数据,增加原先的数值型变量
        dataframe_new['age']=dataframe_age
        dataframe_new['bmi']=dataframe_bmi
        dataframe_new['children']=dataframe_children
        dataframe_new['charges']=dataframe_charges
        return dataframe_new
    
data_lm=dummycoding(data)
data_lm.head()
image.png

上述操作,已经完成了对非数值型数据的虚拟化。然后在回归模型中,可以保留女性/不吸烟/北方 三个变量,作为参照组.

#下面,获取我们需要的因变量,和自变量
from sklearn import linear_model

data_lm_y=data_lm['charges']
data_lm_x1=data_lm.drop(['charges'],axis=1)
data_lm_x2=data_lm.drop(['sex_female'],axis=1)
data_lm_x3=data_lm.drop(['smoker_no'],axis=1)
data_lm_x=data_lm.drop(['region_northeast'],axis=1)
data_lm_x.head()
data_lm_y.head()

处理好自变量以后,接下来进行回归模型拟合

regr=linear_model.LinearRegression()
regr.fit(data_lm_x,data_lm_y)

# 打印截距,保留两位小数
print('Intercept:%.2f'% regr.intercept_)

print('Coefficients:')

# 打印回归系数
print(regr.coef_)

print('Residual sum of squares: %.2f'
      % np.mean((regr.predict(data_lm_x) - data_lm_y) ** 2))

print('Variance score: %.2f' % regr.score(data_lm_x, data_lm_y))

Intercept:-0.00
Coefficients:
[ 4.88218006e-13 -4.88241513e-13 -8.48711097e-12 8.48711085e-12
2.91478729e-13 -1.54068681e-12 -6.52275013e-13 1.28863116e-13
3.07629603e-13 4.38695302e-13 1.00000000e+00]
Residual sum of squares: 0.00
Variance score: 1.00

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,837评论 6 496
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,551评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,417评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,448评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,524评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,554评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,569评论 3 414
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,316评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,766评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,077评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,240评论 1 343
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,912评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,560评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,176评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,425评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,114评论 2 366
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,114评论 2 352

推荐阅读更多精彩内容