机器学习实战--住房月租金预测（3）

PM小王

发布于 2019-07-02 16:00:32

1K0

发布于 2019-07-02 16:00:32

写在前面

这是机器学习系列的第三篇文章，对于住房租金预测比赛的总结这将是最后一篇文章了，比赛持续一个月自己的总结竟然也用了一个月，牵强一点来说机器学习也将会是一个漫长的道路，后续机器学习的文章大多数以知识科普为主，毕竟自己在机器学习这个领域是个渣渣，自己学到的新知识点会分享给大家的。

前文回顾：

从0开始如何用一个月杀进机器学习比赛Top25%

机器学习实战--住房月租金预测（1）

机器学习实战--住房月租金预测（2）

前面的文章谈了谈这次比赛非技术方面的收获，对数据集的初步了解和特征工程的处理，今天主要介绍这次使用的模型--XGBOOST。

XGBOOST模型介绍

关于xgboost的原理网络上的资源很少，大多数还停留在应用层面，自己也是仅仅学习了一点应用，关于原理可以参考陈天奇博士的这篇文章https://xgboost.readthedocs.io/en/latest/tutorials/model.html。

简单介绍：

XGBOOST是一个监督模型，xgboost对应的模型本质是一堆CART树。用一堆树做预测，就是将每棵树的预测值加到一起作为最终的预测值。下图就是CART树和一堆CART树的示例，用来判断一个人是否会喜欢计算机游戏：

第二张图明了如何用一堆CART树做预测，就是简单将各个树的预测分数相加。

参数介绍：

官方参数介绍看这里： https://xgboost.readthedocs.io/en/latest/parameter.html#general-parameters

比较重要的参数介绍：

“reg:linear” –线性回归。 “reg:logistic” –逻辑回归。 “binary:logistic” –二分类的逻辑回归问题，输出为概率。 “binary:logitraw” –二分类的逻辑回归问题，输出的结果为wTx。

“count:poisson”–计数问题的poisson回归，输出结果为poisson分布。在poisson回归中，max_delta_step的缺省值为0.7。(used to safeguard optimization)

“multi:softmax”–让XGBoost采用softmax目标函数处理多分类问题，同时需要设置参数num_class（类别个数）

“multi:softprob” –和softmax一样，但是输出的是ndata * nclass的向量，可以将该向量reshape成ndata行nclass列的矩阵。没行数据表示样本所属于每个类别的概率。

lambda [default=0] L2 正则的惩罚系数 alpha [default=0] L1 正则的惩罚系数

lambda_bias 在偏置上的L2正则。缺省值为0（在L1上没有偏置项的正则，因为L1时偏置不重要）

eta [default=0.3] 为了防止过拟合，更新过程中用到的收缩步长。在每次提升计算之后，算法会直接获得新特征的权重。eta通过缩减特征的权重使提升计算过程更加保守。取值范围为：[0,1]

max_depth[default=6]数的最大深度。缺省值为6，取值范围为：[1,∞]

min_child_weight [default=1] 孩子节点中最小的样本权重和。如果一个叶子节点的样本权重和小于min_child_weight则拆分过程结束。在现行回归模型中，这个参数是指建立每个模型所需要的最小样本数。该成熟越大算法越conservative 取值范围为: [0,∞]

xgboost参数设置的代码示例：

 1xgboost参数设置代码示例：
 2
 3# 划分数据集
 4X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.01, random_state=1729)
 5print(X_train.shape, X_test.shape)
 6
 7#模型参数设置
 8xlf = xgb.XGBRegressor(max_depth=10, 
 9                        learning_rate=0.1, 
10                        n_estimators=10, 
11                        silent=True, 
12                        objective='reg:linear', 
13                        nthread=-1, 
14                        gamma=0,
15                        min_child_weight=1, 
16                        max_delta_step=0, 
17                        subsample=0.85, 
18                        colsample_bytree=0.7, 
19                        colsample_bylevel=1, 
20                        reg_alpha=0, 
21                        reg_lambda=1, 
22                        scale_pos_weight=1, 
23                        seed=1440, 
24                        missing=None)
25
26xlf.fit(X_train, y_train, eval_metric='rmse', verbose = True, eval_set = [(X_test, y_test)],early_stopping_rounds=100)
27
28# 计算 分数、预测
29preds = xlf.predict(X_test)

比赛代码

关于xgboost只是简单的做了一个介绍，自己也仅仅懂一点应用层，原理懂得不是很多,这次XGB代码分析使用的是第一名开源代码。

导入数据集

 1import pandas as pd
 2import numpy as np
 3import matplotlib.pyplot as plt
 4
 5train_data=pd.read_csv('train.csv')
 6test_df=pd.read_csv('test.csv')
 7train_df=train_data[train_data.loc[:,'Time']<3]
 8val_df=train_data[train_data.loc[:,'Time']==3]
 9
10del train_data

以默认参数的XGB分数为准，低于此基准线2.554的模型一律不考虑。

 1def xgb_eval(train_df,val_df):
 2    train_df=train_df.copy()
 3    val_df=val_df.copy()
 4
 5    try:
 6        from sklearn.preprocessing import LabelEncoder
 7        lb_encoder=LabelEncoder()
 8        lb_encoder.fit(train_df.loc[:,'RoomDir'].append(val_df.loc[:,'RoomDir']))
 9        train_df.loc[:,'RoomDir']=lb_encoder.transform(train_df.loc[:,'RoomDir'])
10        val_df.loc[:,'RoomDir']=lb_encoder.transform(val_df.loc[:,'RoomDir'])
11    except Exception as e:
12        print(e)
13
14    import xgboost as xgb
15    X_train=train_df.drop(['Rental'],axis=1)
16    Y_train=train_df.loc[:,'Rental'].values
17    X_val=val_df.drop(['Rental'],axis=1)
18    Y_val=val_df.loc[:,'Rental'].values
19
20    from sklearn.metrics import mean_squared_error
21
22    try:
23        eval_df=val_df.copy().drop('Time',axis=1)
24    except Exception as e:
25        eval_df=val_df.copy()
26
27    reg_model=xgb.XGBRegressor(max_depth=5,n_estimators=500,n_jobs=-1)
28    reg_model.fit(X_train,Y_train)
29
30    y_pred=reg_model.predict(X_val)
31    print(np.sqrt(mean_squared_error(Y_val,y_pred)),end=' ')
32
33    eval_df.loc[:,'Y_pred']=y_pred
34    eval_df.loc[:,'RE']=eval_df.loc[:,'Y_pred']-eval_df.loc[:,'Rental']
35
36    print('')
37    feature=X_train.columns
38    fe_im=reg_model.feature_importances_
39    print(pd.DataFrame({'fe':feature,'im':fe_im}).sort_values(by='im',ascending=False))
40
41    import matplotlib.pyplot as plt
42    plt.clf()
43    plt.figure(figsize=(15,4))
44    plt.plot([Y_train.min(),Y_train.max()],[0,0],color='red')
45    plt.scatter(x=eval_df.loc[:,'Rental'],y=eval_df.loc[:,'RE'])
46    plt.show()
47
48    return eval_df

原生特征的丢弃尝试

以XGB做原生特征筛选，在原生特征中丢弃后不影响分数甚至涨分的特征有：Time，RentRoom(涨幅明显)，RoomDir，Livingroom，RentType(涨幅明显)，SubwayLine(涨幅明显)，SubwayDis(涨幅明显)。

 1# 丢弃各特征后的分数
 2# ‘Time':2.558,'Neighborhood':2.592,'RentRoom':2.531,'Height':2.57,'TolHeight':2.591,'RoomArea':3
 3# 'RoomDir':2.548,'RentStatus':2.561,'Bedroom':2.584,'Livingroom':2.548,'Bathroom':2.590,'RentType':2.538
 4# 'Region':2.583,'BusLoc':2.594,'SubwayLine':2.521,'SubwaySta':2.569,'SubwayDis':2.537,'RemodCond':2.571
 5for col in train_df.columns:
 6    if col!='Rental':
 7        print('drop col:{}'.format(col))
 8        tmp_train_df=train_df.drop([col],axis=1)
 9        tmp_val_df=val_df.drop([col],axis=1)
10        eval_df=xgb_eval(train_df=tmp_train_df,val_df=tmp_val_df)
11
12# 一起丢弃:2.553
13tmp_train_df=train_df.copy()
14tmp_val_df=val_df.copy()
15tmp_train_df.drop(['Time','RentRoom','RoomDir','Livingroom','RentType','SubwayLine','SubwayDis'],axis=1,inplace=True)
16tmp_val_df.drop(['Time','RentRoom','RoomDir','Livingroom','RentType','SubwayLine','SubwayDis'],axis=1,inplace=True)
17eval_df=xgb_eval(train_df=tmp_train_df,val_df=tmp_val_df)

特征选择

一股脑加上所有特征表现不佳，使用贪心策略(前向选择、后向选择)逐个添加特征。

 1train_data=pd.read_csv('train.csv')
 2train_df=train_data[train_data.loc[:,'Time']<3]
 3val_df=train_data[train_data.loc[:,'Time']==3]
 4
 5drop_cols=['SubwayLine','RentRoom','Time']        # 需要丢弃的原生特征
 6
 7comb_train_df=train_df.copy()
 8comb_val_df=val_df.copy()
 9
10# 前向特征选择这块我是用for循环暴力搜出来的最优特征组合，最终筛选出来的特征组合为：
11# ['ab_Height','TolRooms','Area/Room','BusLoc_rank','SubwayLine_rank']
12
13comb_train_df.loc[:,'ab_Height']=comb_train_df.loc[:,'Height']/(comb_train_df.loc[:,'TolHeight']+1)
14comb_val_df.loc[:,'ab_Height']=comb_val_df.loc[:,'Height']/(comb_val_df.loc[:,'TolHeight']+1)
15
16comb_train_df.loc[:,'TolRooms']=comb_train_df.loc[:,'Livingroom']+comb_train_df.loc[:,'Bedroom']+comb_train_df.loc[:,'Bathroom']
17comb_val_df.loc[:,'TolRooms']=comb_val_df.loc[:,'Livingroom']+comb_val_df.loc[:,'Bedroom']+comb_val_df.loc[:,'Bathroom']
18comb_train_df.loc[:,'Area/Room']=comb_train_df.loc[:,'RoomArea']/(comb_train_df.loc[:,'TolRooms']+1)
19comb_val_df.loc[:,'Area/Room']=comb_val_df.loc[:,'RoomArea']/(comb_val_df.loc[:,'TolRooms']+1)
20
21rank_cols=['BusLoc','SubwayLine']
22for col in rank_cols:
23    rank_df=train_df.loc[:,[col,'Rental']].groupby(col,as_index=False).mean().sort_values(by='Rental').reset_index(drop=True)
24    rank_df.loc[:,col+'_rank']=rank_df.index+1        # +1，为缺失值预留一个0值的rank
25    rank_fe_df=rank_df.drop(['Rental'],axis=1)
26    comb_train_df=comb_train_df.merge(rank_fe_df,how='left',on=col)
27    comb_val_df=comb_val_df.merge(rank_fe_df,how='left',on=col)
28    try:
29        comb_train_df.drop([col],axis=1,inplace=True)
30        comb_val_df.drop([col],axis=1,inplace=True)
31    except Exception as e:
32        print(e)
33for drop_col in drop_cols:
34        try:
35            comb_train_df.drop(drop_col,axis=1,inplace=True)
36            comb_val_df.drop(drop_col,axis=1,inplace=True)
37        except Exception as e:
38            pass
39
40# 贪心策略添加特征，目前为:2.403
41eval_df=xgb_eval(train_df=comb_train_df,val_df=comb_val_df

调参对于不是很大的数据集可以用sklearn的Gridcvsearch来暴力调参。示例代码：

1params = {'depth':[3],
2          'iterations':[5000],
3          'learning_rate':[0.1,0.2,0.3],
4          'l2_leaf_reg':[3,1,5,10,100],
5          'border_count':[32,5,10,20,50,100,200]}
6clf = GridSearchCV(cat, params, cv=3)
7clf.fit(x_train_2, y_train_2)

对于较大的数据集，用第一种方法耗时特别长。 2. 逐个参数调，先取定其它参数，遍历第一个参数，选择最优值，再调下一个参数。省时但有的时候容易陷入局部最优。 3.观察数据的分布来调整对应的参数，如树模型的叶子节点数，变量较多，叶子数少欠拟合。

预测提交

 1def xgb_pred():
 2    train_df=pd.read_csv('train.csv')
 3    test_df=pd.read_csv('test.csv')
 4
 5    try:
 6        from sklearn.preprocessing import LabelEncoder
 7        lb_encoder=LabelEncoder()
 8        lb_encoder.fit(train_df.loc[:,'RoomDir'].append(test_df.loc[:,'RoomDir']))
 9        train_df.loc[:,'RoomDir']=lb_encoder.transform(train_df.loc[:,'RoomDir'])
10        test_df.loc[:,'RoomDir']=lb_encoder.transform(test_df.loc[:,'RoomDir'])
11    except Exception as e:
12        print(e)
13
14    train_df.loc[:,'ab_Height']=train_df.loc[:,'Height']/(train_df.loc[:,'TolHeight']+1)
15    test_df.loc[:,'ab_Height']=test_df.loc[:,'Height']/(test_df.loc[:,'TolHeight']+1)
16    train_df.loc[:,'TolRooms']=train_df.loc[:,'Livingroom']+train_df.loc[:,'Bedroom']+train_df.loc[:,'Bathroom']
17    test_df.loc[:,'TolRooms']=test_df.loc[:,'Livingroom']+test_df.loc[:,'Bedroom']+test_df.loc[:,'Bathroom']
18    train_df.loc[:,'Area/Room']=train_df.loc[:,'RoomArea']/(train_df.loc[:,'TolRooms']+1)
19    test_df.loc[:,'Area/Room']=test_df.loc[:,'RoomArea']/(test_df.loc[:,'TolRooms']+1)
20
21    rank_cols=['BusLoc','SubwayLine']
22    for col in rank_cols:
23        rank_df=train_df.loc[:,[col,'Rental']].groupby(col,as_index=False).mean().sort_values(by='Rental').reset_index(drop=True)
24        rank_df.loc[:,col+'_rank']=rank_df.index+1        # +1，为缺失值预留一个0值的rank
25        rank_fe_df=rank_df.drop(['Rental'],axis=1)
26        train_df=train_df.merge(rank_fe_df,how='left',on=col)
27        test_df=test_df.merge(rank_fe_df,how='left',on=col)
28        try:
29            train_df.drop([col],axis=1,inplace=True)
30            test_df.drop([col],axis=1,inplace=True)
31        except Exception as e:
32            print(e)
33    for drop_col in drop_cols:
34            try:
35                train_df.drop(drop_col,axis=1,inplace=True)
36                test_df.drop(drop_col,axis=1,inplace=True)
37            except Exception as e:
38                pass
39
40    print(train_df.columns,test_df.columns)
41
42    import xgboost as xgb
43    X_train=train_df.drop(['Rental'],axis=1)
44    Y_train=train_df.loc[:,'Rental'].values
45    test_id=test_df.loc[:,'id']
46    X_test=test_df.drop(['id'],axis=1)
47
48
49    from sklearn.metrics import mean_squared_error
50
51    reg_model=xgb.XGBRegressor(max_depth=8,n_estimators=1880,n_jobs=-1)
52    reg_model.fit(X_train,Y_train,eval_set=[(X_train,Y_train)],verbose=100,early_stopping_rounds=10)
53
54    y_pred=reg_model.predict(X_test)
55
56    sub_df=pd.DataFrame({
57        'id':test_id,
58        'price':y_pred
59    })
60    sub_df.to_csv('sub.csv',index=False)
61
62    return None
63
64xgb_pred()

第一名XGB单模分数为1.94，线下线上是一致的，总特征数二十多个，跟自己的XGB相比，自己在特征组合方向有所欠缺，自己单模特征10个左右分数在2.01。

本文参与腾讯云自媒体分享计划，分享自微信公众号。

原始发表：2019-01-12，如有侵权请联系 cloudcommunity@tencent.com 删除

深度学习