房价精准预测,大数据+机器学习帮你搞定

一提到房价,就“压力山大”!无论是首套房还是改善性需求,买在低点卖在高点都是一个可遇不可求的事儿,所以如果有位数据大侠能帮助设计一个预测房价的神器,岂不是“人生很值得”!本期DT数据侠与纽约数据科学学院合作的数据线专栏中,四位数据侠通过“数据超能力”试图利用Python通过机器学习方式来预测房价,快来看看他们是如何做得吧!

如果让你全凭直觉来判断,上图里的四个房子哪个最贵?

(也许)大多数人会说是右边的蓝色房子,因为它看起来最大也最新。然而,当你看完今天这篇文章,你可能会有不同的答案,并且发现一种更准确的预测房屋价格的方法。

这个项目的数据集可以在kaggle页面(https://www.kaggle.com/c/house-prices-advanced-regression-techniques)找到:这些数据被分为两类,训练集和测试集。

数据列表一共有2600行、79列,包括了不同房屋的描述性数据,比如卧室数、一层的房屋面积等。训练集里还包括了房屋的真实价格数据。

▍因变量

(图片说明:房价的分布图)

房屋价格总体来看,是一个平均值和中位数在20万美元左右的向右倾斜的分布,最高的价格在55万到75万之间。

(图片说明:房价的描述统计)

▍自变量

类别变量(Categorical Variables )

(图片说明:房屋质量vs.房价)

大多数(79种变量中有51种)是定性变量(categorical),包括房子所在社区、整体质量、房屋类型等。最好预测的变量是与质量相关的变量。比如,整体质量这个变量最终证明是预测价格的最关键因素。房子某一个部分的质量,比如泳池、地下室等,也都与最终价格有很强的相关性。

数字变量(numeric variables)

数字变量大多是关于房屋面积。它们也与价格相关。

(图片说明:除去地下室的房屋居住面积vs.房价)

(图片说明:总面积vs.房价)

▍缺失的数据

(图片说明:缺失值)

过程中的一大挑战是那些缺失的数据,对于像泳池质量、泳池面积等数据,如果数据缺失了,则说明这栋房子没有泳池,我们会用0来代替,如果是定性变量,则用“无”来代替。对于那些“意外缺失”的数据,我们则通过其他变量进行估算,补充进去。

(图片说明:缺失值与填补)

▍特征工程

处理一大堆不清晰的特征总是充满挑战。下面我们要创造和抛弃一些变量,并引入一些哑变量等。

抛弃变量

通常人们会删除一些相互高度关联的特征。在我们的分析中,我发现车库建造年份和房屋建造年份关联度很强,关联值达到0.83。而且75.8%以上的情况下,这两个值是相同的。因此,我们决定把有很多缺失的车库年份数据丢掉。

创造新的变量

有时候需要创造新的变量从而提升整个模型的表现,我们设计了两个新变量:

1. 卖掉时的房龄

2. 卖掉时距重新装修过去多少年

▍处理变量

1. 我们找出11个定性变量,它们存在某种排序的可能,可以分别将它们划分为很棒、一般和很差;

2. 对于其他的定性变量,我们使用pandas.get_dummies来得到独热编码(One-Hot Encoding);

3. 我们找到24个连续数据变量,它们的斜率大于0.75(向右倾斜),我们使用对数变换来去掉本身的偏态。

(图片说明:特征检测)

▍正则化(regularization)

因为我们需要处理很多变量,所以我们引入了正则化的操作,来处理在过程中发现的那些多重共线性关系,以及使用多元线性回归模型可能带来的过度拟合问题。

正则化最棒的地方在于它能减少模型的复杂性,因为它能自动地为你完成特征挑选的任务。所有正则化模型都会惩罚多余的特征。

正则化的模型包括 Lasso、Ridge 模型和弹性网络(Elastic Net)。Lasso 算法(最小绝对值收敛和选择算法)会将系数设为0,而ridge回归模型会最小化系数,使其中的一些非常接近0。弹性网络模型是Lasso和Ridge的混合。它将彼此相关的变量分到同一组,如果里面有一个变量是个很强的预测变量(predictor),那么整个组都会被纳入这个模型。

下一步是将每个模型的超参数进行交叉验证。

我们将Lasso模型的阿尔法定为 = .0005,Ridge的阿尔法为2.8 。弹性网络模型的阿尔法为 .0005 , L1_Ratio = 0.9。因为当 L1_Ratio = 0.9 时,弹性网络模型十分接近 Lasso模型,后者有默认的 L1_Ratio = 1 。

▍特征选择

(图片说明:Lasso与Ridge模型的回归系数)

Lasso模型

对房屋价格的正算子系数:地上生活空间、整体房子状况以及Stone Bridge、North Ridge 和 Crawford社区。

负算子系数:MS Zoing、Edwards 社区和地上厨房。

Ridge模型

对房屋价格的正算子系数:整体住宅面积、房顶材料(木瓦)、整体状况。

负算子系数:一般的分区需求、离主干道或铁路的距离,以及游泳池状况良好。

▍训练数据中模型预测的价格和真实价格的对比

下面两图展示了我们的模型的精确度。离红线近的以及在红线上面的是我们预测准确的,那些偏离的比价多的需要我们进一步研究。

(图片说明:预测价格vs.真实价格)

▍梯度提升回归(Gradient boosting regression)

梯度提升回归是我们表现最好的一个算法。我们最初使用全部特征(基准模型)来训练梯度提升机。我使用 scikit-learn这个Python包提供的 GridSearchCV 功能来进行参数调整的交叉验证。我们最好的模型参数是:学习值0.05,估计量2000,最大深度3。

我们制作了一个相对重要性表格,将梯度提升特征的重要性用可视化的方式呈现。特征重要性分数代表每个特征在构建这个加强版的决策树里是否有用。地上生活空间面积、厨房质量、地下室面积以及车库大小是最重要的特征。

(图片说明:相对重要性)

▍PCA(主成分分析 )+ 梯度提升回归

我们接下来尝试通过减少特征的维度来提高我们基准模型的表现。高维度的数据可能很分散,就会让使用某种算法来训练有用的模型变得更难。总的来说,最优的、非多余的特征子集会对预测性的算法有好处,能够提高训练率以及加强它的可解释性和一般性。

我们使用 scikit-learn 的 Pipelines 来管理我们的机器学习模型,它允许我们通过应用一个估计量来完成一系列数据的转化工作。

我们设计了不同的pipeline,每一个有不同的估计量。对于梯度提升回归,我们的pipelin包括:

1. 特征缩放,使用了scikit-learn 的python包

2. 降维,使用PCA(留下了150个主要的成分)

我们完成了特征工程后,得到200个特征和大约1500行训练数据集。在看过累积方差的百分比的表格后,我们决定留下150个核心元素。

(图片说明:方差累积百分比)

并不是所有调整都能优化结果。在我们用PCA操作后,交叉验证的分数并没有提高,甚至恶化了(从0.91降到了0.87)。我们相信是降维时,也去掉了一些关键信息。PCA 不仅去掉了随机出现的噪音,也去掉了有价值的信息。

▍PCA + 多元线性回归

对于多元线性回归,我们的pipeline 包括:

1. 特征缩放,使用了scikit-learn 的python包

2. 降维,使用PCA(留下了150个主要的成分)

使用多元线性回归的PCA 也没有带来好的结果。交叉验证的分数并没有提高,甚至恶化了。

▍模型比较

XG Boost 是表现最好的模型,多元线性回归表现最差,其他模型的结果差不多。

(图片说明:各个模型对比)

使用单独某一个模型都能让我们得到不错的结果。但是,通常来说,真实生活中的问题并没有一种线性或者非线性的关系,可以让我们用一个单独的模型来重现。把保守和激进、线性和非线性的模型结合起来,才能最好地呈现房价预测这个问题。

▍融合(stacking 和 ensembling)

我们先尝试了一个简单的合模型(ensembling),以50-50的比例将 XGBoost(非线性)和ENet(线性)组合在一起。

接下来,我们按照模型融合(stacking)的基本方法,又尝试了多个不同模型,来看哪个效果最好。下图记录了这些不同模型的表现情况。

▍结论

下面的相关性热点图展示了我们不同模型的预测价格。可以看到,弹性网络、Lasso和Ridge本质上很相似,而两种融合方式也彼此很像。与其他都明显不同的是 XGBoost 模型。

(图片说明:模型成绩)

▍未来可研究方向

1. 研究自变量之间的相关性

2. 尝试更多的特征工程

3. 使用聚类分析来创造更多新的特征

4. 对不同模型使用不同的特征选择方法:在线性模型中抛弃掉特定的特征,而在树形模型中保持大多数的特征。

注:本文编译自纽约数据科学院博客《Predicting House Prices with Machine Learning Algorithms》,点击“阅读原文”查看。内容仅为作者观点,不代表DT数据侠立场。文中图片部分来自作者。

作者 | Yulia Norenko等

题图 | 站酷海洛

期待更多数据侠干货分享、话题讨论、福利发放?在公众号DT数据侠(ID:DTdatahero)后台回复“数据社群”,可申请加入DT数据社群。

▍关于DT×NYCDSA

DT×NYCDSA是DT财经与纽约数据科学学院合作专栏。纽约数据科学学院(NYC Data Science Academy)是由一批活跃在全球的数据科学、大数据专家和SupStat Inc. 的成员共同组建的教育集团。

▍数据侠门派

本文数据侠Yulia Norenko ,有5年的金融从业经验,曾在多家知名券商担任分析师职务。他拥有纽约大学和莫斯科大学的硕士学位。

本文数据侠Howard Chang,毕业于石溪大学数学专业。他曾在管理数十亿美元的对冲基金工作。

本文数据侠Lavanya Gupta,毕业于安巴尼信息与通信技术研究所,目前就职于汇丰银行,担任软件工程师。

本文数据侠Silvia Lu,目前在纽约和新泽西地区空港管理部门实习,担任数据分析师职务,她拥有纽约大学的心理学专业硕士学位。


那些你可能错过的干货

▍加入数据侠

数据侠计划是由第一财经旗下DT财经发起的数据社群,包含数据侠专栏、数据侠实验室系列活动和数据侠联盟,旨在聚集大数据领域精英,共同挖掘数据价值。申请入群请添加DT君微信(dtcaijing003)并备注“数据社群”,合作请联系datahero@dtcj.com。

原文发布于微信公众号 - DT数据侠(DTdatahero)

原文发表时间:2018-11-16

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏新智元

【深度学习路线图】关键概念、模型及其发展关系

【新智元导读】新智元以前也译介过 Carlos E. Perez 关于深度学习的文章。这次我们要介绍的是他最新制作的“深度学习路线图”。 Perez 发现,他...

7244
来自专栏AI科技评论

机器学习顶会 ICML 2018 斯德哥尔摩开幕,公平性研究戴上光环

AI 科技评论按:经过 7 月 10 日的三个 Tutorial Session 和 Opening Reception 晚宴之后,在瑞典斯德哥尔摩举办的机器学...

1494
来自专栏计算机视觉战队

未来人工智能之人脸领域技术

部分来源于《机器人大讲堂》和《2017年中国人脸识别未来发展路径、市场需求、市场发展空间预测》 ? 近年来由于深度学习爆炸式的发展,已经带动了整个行业的发展。...

3265
来自专栏量子位

AI学会了视觉推理,“脑补”看不清的物体 | 李佳李飞飞等的CVPR论文

这张照片在人类眼中,左下角的一小片棕黄黑灰就是一匹活生生的马——这个姿势和场景,当然是骑马啦!

1494
来自专栏新智元

【干货】机器学习工程师必须知道的十个算法

【新智元导读】机器学习算法可以分为三个大类:监督学习、无监督学习、强化学习。监督学习对于有属性(标记)的特定数据集(训练集)是非常有效的。无监督学习对于在给定未...

3076
来自专栏量子位

手把手带你入坑迁移学习(by 当过黑客的CTO大叔)

原作 Slav Ivanov Root 编译自 Slav寄几的博客 量子位 出品 | 公众号 QbitAI Slav Ivanov是Post Planer(提高...

3396
来自专栏用户2442861的专栏

结合实例与代码谈数字图像处理都研究什么?

图像处理(以及机器视觉)在学校里是一个很大的研究方向,很多研究生、博士生都在导师的带领下从事着这方面的研究。另外,就工作而言,也确实有很多这方面的岗位和机会虚...

2591
来自专栏机器之心

学界 | 结合生成式与判别式方法,Petuum新研究助力医疗诊断

36412
来自专栏AI科技评论

Getmax 团队:KDD CUP 2018 两冠一亚团队解题思路

AI 科技评论按:今年 KDD CUP 设立三项大奖,分别为 General Track、Last Ten-Day Prediction Track 以及 Se...

1712
来自专栏京东技术

京东大数据研发部3篇论文同时被国际顶级人工智能协会AAAI收录

京东大数据研发部与中科院、美国西北大学、美国加州大学伯克利分校等相关机构合作,深度参与技术研究,在即将举行的第三十二届美国人工智能顶级国际会议AAAI-2018...

40311

扫码关注云+社区

领取腾讯云代金券