【火炉炼AI】机器学习006-用决策树回归器构建房价评估模型

【火炉炼AI】机器学习006-用决策树回归器构建房价评估模型

(本文所使用的Python库和版本号: Python 3.5, Numpy 1.14, scikit-learn 0.19, matplotlib 2.2 )

最近几十年,房价一直是中国老百姓心中永远的痛,有人说,中国房价就像女人的无肩带文胸,一半人在疑惑:是什么支撑了它?另一半人在等待:什么时候掉下去? 而女人,永不可能让它掉下来。就算快掉下来了,提一提还是又上去了.....

虽然我们不能预测中国房价什么时候崩盘,但是却可以用机器学习来构建房价预测模型,下面我们使用波士顿房价数据集来建立一个房价评估模型,通过房子所在的地理位置,人口居住特性等因素来综合评估房价。


1. 分析数据集

本次模型所使用的数据集来自于波士顿房价数据集官方网站,地址为:(http://www.cs.toronto.edu/~delve/data/boston/bostonDetail.html)。

这个数据集比较小,只有506个样本,每个样本有13个features, 1个label,关于这14个属性的说明如下图所示:

image

所以我们需要构建模型,使用前面的13个特征来评估最后的MEDV,即使用13个特征向量来计算房价,这是典型的多元回归问题。虽然可以从官网上直接下载这个数据集,但是sklearn中已经集成了该数据集,我们可以直接调用。如下代码

# 分析数据集
from sklearn import datasets # sklearn自带的datasets中就有Boston房价数据集
housing_data=datasets.load_boston()
dataset_X=housing_data.data # 获取影响房价的特征向量,作为feaure X
dataset_y=housing_data.target # 获取对应的房价,作为label y
# print(dataset_X.shape) # (506, 13) # 一共有506个样本,每个样本有13个features
# print(dataset_y.shape) # (506,)
# print(dataset_X[:5,:]) # 打印看看features的数值类型和大小,貌似已经normalize.

# 将整个数据集划分为train set 和test set两部分
from sklearn.utils import shuffle
dataset_X,dataset_y=shuffle(dataset_X,dataset_y)
# print(dataset_X[:5,:]) # 确认dataset_X 的确发生了shuffle
num_split=int(0.8*len(dataset_X))
train_X,train_y=dataset_X[:num_split],dataset_y[:num_split]
test_X,test_y=dataset_X[num_split:],dataset_y[num_split:]
# print(train_X.shape) # (404, 13)
# print(test_X.shape) # (102, 13)

# 上面的数据集划分也可以采用下面的方法:
# from sklearn.model_selection import train_test_split
# dataset_y=dataset_y[:,np.newaxis]
# dataset=np.hstack((dataset_X,dataset_y))
# print(dataset.shape)
# print(dataset[:,:3])
# train_set,test_set=train_test_split(dataset,test_size=0.2,random_state=37)
# print(train_set.shape) # (404, 14)
# print(test_set.shape) # (102, 14)

上述代码首先调用sklearn.load_boston()来获取波士顿房价数据集,然后将该数据集划分为两部分,其中train set占据80%(即404个样本),test set占据20%(102个样本)。在查看数据集中前面五行的结果时,发现整个数据集已经Normalize,故而此处我们没有必要进行归一化。


2. 构建决策树模型

决策树(DecisionTree, DT)是一种常见的用于分类和回归的非参数监督学习方法,目标是创建一个模型,通过从数据特性中推导出简单的决策规则来预测目标变量的值。

决策树模型的优点在于:1,简单容易理解,数结构可以可视化表达。2,需要很少的数据准备,其他技术通常需要数据标准化,需要创建虚拟变量,并删除空白值。3,能够处理多输出问题。4,相比于使用神经网络的模型,决策树是白盒模型,结果很容易解释。

决策树模型的缺点在于:1,决策树学习可能会生成过于复杂的数结构,不能代表普遍的规则,即模型容易过拟合,修剪机制,设置叶子节点所需的最小样本数目或设置数的最大深度是避免决策树过拟合的必要条件。2,决策树可能不稳定,因为数据中的小变化可能导致生成完全不同的树。这个问题可以通过在一个集合中使用多个决策树来减轻。3,实际的决策树学习算法是基于启发式算法的,例如在每个节点上进行局部最优决策的贪婪算法。这种算法不能保证返回全局最优决策树。通过在集合学习中训练多个树,可以减少这种情况,在这里,特征和样本是随机抽取的。4,有些概念很难学习,因为决策树无法很容易地表达它们,例如XOR、奇偶性或多路复用问题。

说了这么多废话,直接看代码吧。。。

# 构建决策树回归模型
from sklearn.tree import DecisionTreeRegressor
decision_regressor=DecisionTreeRegressor(max_depth=4) # 最大深度确定为4
decision_regressor.fit(train_X,train_y) # 对决策树回归模型进行训练

# 使用测试集来评价该决策树回归模型
predict_test_y=decision_regressor.predict(test_X)

import sklearn.metrics as metrics
print('决策树回归模型的评测结果----->>>')
print('均方误差MSE:{}'.format(
    round(metrics.mean_squared_error(predict_test_y,test_y),2)))
print('解释方差分:{}'.format(
    round(metrics.explained_variance_score(predict_test_y,test_y),2)))

-------------------------------------输---------出--------------------------------

决策树回归模型的评测结果----->>>
均方误差MSE:13.33
解释方差分:0.81

--------------------------------------------完-------------------------------------

########################小**********结###############################

1. 决策树回归模型的构建非常简单,和线性回归器的构建类似,使用DecisionTreeRegressor获取一个回归器对象即可,模型的训练使用fit函数,预测使用predict函数即可。

2. 简单的决策树回归模型得到的MSE比较大,解释方差分结果也不理想,说明这个模型还有很大的优化空间。

#################################################################


3. 决策树模型的优化

一般的,可以先从数据集上优化,但本项目的数据集是非常成熟的案例,进一步优化的空间不大,所以只能从模型上下手。此处我采用两种优化方法,第一种是优化决策树模型中的各种参数,比如优化max_depth,判断是否可以改进MSE,第二种方式是使用AdaBoost算法来增强模型的准确性。

# 第一种优化方法:改变max depth来降低MSE,提高解释方差分
for depth in range(2,12):
    decision_regressor_test=DecisionTreeRegressor(max_depth=depth) 
    decision_regressor_test.fit(train_X,train_y)
    predict_test_y2=decision_regressor_test.predict(test_X)
    print('depth: {}, MSE: {:.2f}, EVS: {:.2f}'.format(
        str(depth), metrics.mean_squared_error(predict_test_y2,test_y),
        metrics.explained_variance_score(predict_test_y2,test_y)))

-------------------------------------输---------出--------------------------------

depth: 2, MSE: 23.83, EVS: 0.62
depth: 3, MSE: 20.98, EVS: 0.68
depth: 4, MSE: 13.33, EVS: 0.81
depth: 5, MSE: 13.00, EVS: 0.83
depth: 6, MSE: 14.36, EVS: 0.83
depth: 7, MSE: 11.73, EVS: 0.86
depth: 8, MSE: 16.18, EVS: 0.83
depth: 9, MSE: 15.74, EVS: 0.83
depth: 10, MSE: 14.67, EVS: 0.83
depth: 11, MSE: 15.02, EVS: 0.84

--------------------------------------------完-------------------------------------

# 第二种优化方法:通过AdaBoost算法来提高模型准确度
from sklearn.ensemble import AdaBoostRegressor
ada_regressor=AdaBoostRegressor(DecisionTreeRegressor(max_depth=7),n_estimators=400)
ada_regressor.fit(train_X,train_y)

# 查看使用AdaBoost算法后模型的MSE 和EVS
predict_test_y=ada_regressor.predict(test_X)

import sklearn.metrics as metrics
print('AdaBoost决策树回归模型的评测结果----->>>')
print('均方误差MSE:{}'.format(
    round(metrics.mean_squared_error(predict_test_y,test_y),2)))
print('解释方差分:{}'.format(
    round(metrics.explained_variance_score(predict_test_y,test_y),2)))

-------------------------------------输---------出--------------------------------

AdaBoost决策树回归模型的评测结果----->>>
均方误差MSE:7.87
解释方差分:0.9

--------------------------------------------完-------------------------------------

########################小**********结###############################

1,两种优化方法:第一种通过优化max_depth可以发现,在depth=7时得到的MSE最小,且解释方差分最大,故而认定max_depth=7.

2,第二种通过AdaBoost增强算法来优化模型,得到的MSE(7.87)比depth优化最好的MSE(11.73)要小很多,且解释方差分也有了较大提高,说明优化效果比较好。

3,AdaBoost(Adaptive boosting)算法是一种自适应的利用其他系统来增强模型准确性的算法,将不同版本的算法结果进行组合,用加权汇总的方式获得最终结果。AdaBoost算法在每个阶段获取的信息都会反馈到模型中,这样学习器就可以在后一阶段重点训练难以分类的样本。

4,当然,对于本模型,还可以继续优化,比如优化决策树的结构,优化决策树的其他参数等。

#################################################################


4. 特征的相对重要性

本项目共有13个特征,但这13个特征对于房价的影响肯定是不同的,比如从直观上来看,学区房对房价影响肯定比较大,所以可以计算出各种不同特征对模型的影响,进而在特征比较复杂的情况下,可以忽略掉影响比较小的特征。

如下两部分代码可以将各种特征的相对重要性绘制到图中。

# 计算不同特征的相对重要性
def plot_importances(feature_importances, title, feature_names):
    '''将feature_importance绘制到图表中,便于观察,
    并把重要性大于5的特征打印出来'''
    # 将重要性都归一化为0-100之内
    feature_importances=100.0*(feature_importances/max(feature_importances))
    
    # 将得分从高到低排序
    index_sorted=np.flipud(np.argsort(feature_importances))
    # 让X坐标轴上的标签居中显示
    pos=np.arange(index_sorted.shape[0])+0.5
    
    # 画条形图
    plt.figure()
    plt.bar(pos,feature_importances[index_sorted],align='center')
    plt.xticks(pos,feature_names[index_sorted])
    plt.ylabel('Relative Importance')
    plt.title(title)
    
    # 把重要性结果打印出来
    print('{} importance list------>>>>>'.format(title))
    for importance,name in zip(feature_importances[index_sorted],
                               feature_names[index_sorted]):
        if importance>5:
            print('feature:{}, importance: {:.2f}'.format(name,importance))
decision_regressor7=DecisionTreeRegressor(max_depth=7) # 最大深度确定为7
decision_regressor7.fit(train_X,train_y) # 对决策树回归模型进行训练
plot_importances(decision_regressor7.feature_importances_,
                 'DT regressor',housing_data.feature_names)
plot_importances(ada_regressor.feature_importances_,
                 'AdaBoost Optimized DT regressor',housing_data.feature_names)

-------------------------------------输---------出--------------------------------

DT regressor importance list------>>>>>
feature:LSTAT, importance: 100.00
feature:RM, importance: 52.24
feature:DIS, importance: 15.97
feature:PTRATIO, importance: 5.00
AdaBoost Optimized DT regressor importance list------>>>>>
feature:LSTAT, importance: 100.00
feature:RM, importance: 46.72
feature:DIS, importance: 21.80
feature:TAX, importance: 7.34
feature:AGE, importance: 7.29
feature:CRIM, importance: 6.51
feature:NOX, importance: 5.32
feature:PTRATIO, importance: 5.29

--------------------------------------------完-------------------------------------

决策树回归器得到的特征重要性图
AdaBoost优化后得到的特征重要性图

########################小**********结###############################

1,通过上面两幅图的比较,可以看出,各种特征对模型的相对重要性变化很大,两种模型都发现最重要的特征是LSTAT(这和书本《Python机器学习经典实例》中不一样。

2,但AdaBoost增强算法貌似能够提升其他小特征的相对重要性,而且两种模型中特征重要性排序也不一样。

#################################################################


注:本部分代码已经全部上传到(我的github)上,欢迎下载。

参考资料:

1, Python机器学习经典实例,Prateek Joshi著,陶俊杰,陈小莉译

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

推荐阅读更多精彩内容