当前位置:   article > 正文

数据正态分布化 + 基于回归算法的房价预测(含python代码)_xgb回归

xgb回归

本文用到的美国房屋数据,数据介绍详见我的上一篇文章:

链接:https://pan.baidu.com/s/1wrkzFF87A_Emgid_s7K3aA
提取码:2j77 

内含两个文件:

data_train.csv:训练集数据,包含房价等81个指标;

data_test.csv:测试集数据,不包含房价;

文章目录

前言

一、数据预处理

1. 删除异常值

2. 填补缺失值

3. 转换数据类型

4. 建立值与类别的映射 

二、对房价进行正态分布化处理:

         第一步:在原数据的柱状图上添加正态分布概率密度曲线和核密度估计曲线;

         第二步:对房价进行正态分布化处理;

         第三步:添加一列总面积并计算偏度;

         第四步:随后对偏度大的列数据进行BOX-COX转换,改善数据的正态性和对称性;

         第五步:对数据进行独热编码;

三、使用多个回归预测算法进行房价预测

1. 导入包和数据

2. 定义模型

3. 定义训练数据和计算模型均方根误差的方法

4. 通过均方误差评估各模型效果

5. 模型应用(预测)


前言

紧接上一篇文章,本文针对数据进行了正态化处理,应用了Lasso、KernelRidge、ElasticNet、GradientBoostingRegressor、XGBRegressor等回归模型进行房价预测。

数据长这样,数据集的具体解释和分析详见我的上一篇文章


一、数据预处理

导入包:

  1. #不显示警告
  2. import warnings
  3. warnings.filterwarnings('ignore')
  4. #数据处理,数据分析
  5. import numpy as np
  6. import pandas as pd
  7. #统计计算相关的工具包
  8. import math
  9. from scipy import stats
  10. from scipy.stats import norm
  11. #画图相关的工具包
  12. import seaborn
  13. import matplotlib.pyplot as plt
  14. from matplotlib.font_manager import FontProperties
  15. %matplotlib inline

1. 删除异常值

  1. df_train = pd.read_csv('data_train.csv',encoding='gbk')#读入数据文件
  2. # 统计各属性的缺失值个数,并按照降序排序
  3. total_isnull = df_train.isnull().sum().sort_values(ascending=False)
  4. # 统计各行总数
  5. total_number_rows=len(df_train)
  6. # 统计缺失值占比
  7. percent = (total_isnull/total_number_rows)*100
  8. # 拼接两列
  9. missing_data = pd.concat([total_isnull, percent], axis=1, keys=['缺失值数量', '百分比'])
  10. # 查看前20行
  11. missing_data.head(20)

  1. # 读取训练集数据
  2. train = pd.read_csv('data_train.csv',encoding='gbk')
  3. # 读取测试集数据
  4. test = pd.read_csv('data_test.csv',encoding='gbk')
  5. # 查看训练集数据形状
  6. print("The train data size before dropping Id feature is : (%d %d)" %(train.shape[0],test.shape[1]))
  7. # 查看测试集数据形状
  8. print("The test data size before dropping Id feature is : (%d %d)\n" %(test.shape[0],test.shape[1]))
  9. # 删除两个数据集中编号属性列
  10. train.drop("编号", axis = 1, inplace = True)
  11. test.drop("编号", axis = 1, inplace = True)

查看离群点,作居住面积与房价的散点图,发现右下角存在离群点。 

  1. # 设置图片样式
  2. plt.figure(figsize=(16, 8),dpi=600)
  3. # 绘制“住房面积” “房价散点图”
  4. fig = seaborn.scatterplot(train['居住面积'],train['房价'])

删除离群点,再次作居住面积与房价的散点图,观察离群点是否已经被删除。 

  1. train = train.drop(train[(train['居住面积']>4000) & (train['房价']<300000)].index)#删除离群点
  2. plt.figure(figsize=(16, 8),dpi=600)#设置图片大小
  3. fig = seaborn.scatterplot(train['居住面积'],train['房价'])#作散点图

2. 填补缺失值

  1. #计算所有特征缺失值占比
  2. all_data_na = (all_data.isnull().sum() / len(all_data)) * 100
  3. #去掉不包含缺失值的属性,并对剩下的属性按照缺失值比例降序排序,存下排名前10的属性
  4. all_data_na = all_data_na.drop(all_data_na[all_data_na == 0].index).sort_values(ascending=False)[:10]
  5. #转换成DataFrame的格式
  6. missing_data = pd.DataFrame({'缺失值占比' :all_data_na})
  7. #展示
  8. missing_data

填补缺失值。根据特征的现实意义,采用”None“进行填充:

  1. cols = ('泳池质量','车库类型', '车库室内装修', '车库质量', '车库条件','地下室的高度','地下室的条件','花园层地下室墙',
  2. '地下室基底质量','地下室第二成品区质量','砌体饰面型','砌体饰面型','杂项功能','通道入口的类型',
  3. '栅栏的质量','壁炉质量','建筑类','实用工具') #将需要转换的列存入元组
  4. for col in cols:
  5. all_data[col] = all_data[col].fillna('None')
  6. all_data['泳池质量']

  1. #先利用groupby函数将数据按照“街区”进行分组,取出‘街道连接距离’这一列
  2. list(all_data.groupby('街区')['街道连接距离'])[0:3]
  3. cols = ('车库修筑年份','车库规模','车库车位数量','地下室基底面积','地下室第二成品区面积',
  4. '未使用的地下室面积','地下室总面积','地下室全浴室','地下室半浴室','砌体饰面面积','砌体饰面面积') #将需要转换的列存入元组
  5. for col in cols:
  6. all_data[col] = all_data[col].fillna(0)
  7. cols = ('一般分区','电气系统','厨房的品质','第一外部材料','第二外部材料','家庭功能评级','销售类型') #将需要转换的列存入元组
  8. for col in cols:
  9. all_data[col] = all_data[col].fillna(all_data[col].mode()[0])

此时再检查缺失值,发现已经全部填补完全:

  1. all_data_na = (all_data.isnull().sum() / len(all_data)) * 100#计算缺失值比例
  2. all_data_na = all_data_na.drop(all_data_na[all_data_na == 0].index).sort_values(ascending=False)#去掉不包含缺失值的属性并对剩下的属性排序
  3. missing_data = pd.DataFrame({' 缺失值占比' :all_data_na})#转换成数据框格式
  4. missing_data.head()#显示前几行

3. 转换数据类型

转换数据类型,方便后期操作:

  1. all_data['建筑类'] = all_data['建筑类'].apply(str)#将数据转成字符串的形式
  2. all_data['总体状况评价'] = all_data['总体状况评价'].apply(str)
  3. all_data['销售年'] = all_data['销售年'].apply(str)
  4. all_data['销售月'] = all_data['销售月'].apply(str)

4. 建立值与类别的映射 

  1. from sklearn.preprocessing import LabelEncoder
  2. cols = ('壁炉质量', '地下室的高度', '地下室的条件', '车库质量', '车库条件',
  3. '外部材料质量', '外部材料条件','加热质量', '泳池质量', '厨房的品质', '地下室基底质量',
  4. '地下室第二成品区质量', '家庭功能评级', '栅栏的质量', '花园层地下室墙', '车库室内装修', '坡的财产',
  5. '财产的类型', '车道', '道路类型', '通道入口的类型', '中央空调', '建筑类', '总体状况评价',
  6. '销售年', '销售月')#将需要转换的列存入元组
  7. for c in cols:
  8. myLabel = LabelEncoder()#创建labelEncode对象
  9. myLabel.fit(list(all_data[c].values))#建立值与类别的映射关系
  10. all_data[c] = myLabel.transform(list(all_data[c].values))#将每一个值按照映射关系进行转换
  11. print('Shape all_data: (%d,%d)'%(all_data.shape[0],all_data.shape[1]))

二、对房价进行正态分布化处理:

第一步:在原数据的柱状图上添加正态分布概率密度曲线和核密度估计曲线;

  1. (mu, sigma) = norm.fit(train['房价'])#计算变换前的均值和方差
  2. print("\n mu = %.2f and sigma = %.2f\n" %(mu,sigma))#打印均值和方差
  3. #画出变换前房价的核密度估计曲线,拟合出的生态分布密度概率曲线
  4. plt.figure(figsize=(16, 8),dpi=600)#设置图片大小,分辨率
  5. plt.ylabel('概率密度')#设置纵轴标签
  6. plt.title('房价分布')#设置图片标题
  7. #第一个参数为房价数据,fit=norm代表绘制生态分布密度概率曲线(黑色曲线),kde=True代表绘制核密度估计曲线(蓝色曲线)
  8. #fit_kws,kde_kws,hist_kws分别可以设置黑色曲线,蓝色曲线和直方图的图例标签。
  9. seaborn.distplot(train['房价'] ,fit=norm,kde=True,fit_kws={"label":"正态分布概率密度曲线"},
  10. kde_kws={"label":"核密度估计曲线"},hist_kws={"label":"房价频数"})#绘图
  11. plt.legend()#设置图例
  12. plt.figure(figsize=(16, 8),dpi=600)#设置图片大小
  13. res = stats.probplot(train['房价'], plot=plt)#计算概率图的分位数,plt是绘图对象,plot=plt显示该图

 

上述结果解释: 第一张图中的蓝色直方图代表对应区间房价的频数,蓝色曲线为和核密度估计曲线,黑色的曲线代表以房价的均值和方差拟合出的正态分布概率密度曲线。可以看出,观测数据的分布与正态分布有一定差距,故应对观测数据进行正态分布化处理。

第二张图为Q—Q图。

蓝色的作图方法:1、找出标准正态分布的所有分位点作为X轴刻度值;2、找出与X轴上所有分位点对应的样本分位点,以标准正态分布分位点为横坐标,样本分位点为纵坐标,可以做出Q-Q图中蓝色的点;

红色线的作图方法:1、以样本的均值和方差生成一组正态分布样本(对应第一张图中黑色曲线);2、找出与X轴上所有分位点对应的正态分布样本的分位点,以标准正态分布分位点为横坐标,正态分布分位点为纵坐标,可以做出Q-Q图红色直线。

结论:蓝色的点越贴近直线,越说明观测数据的分布贴近正态分布

对房价进行正态分布化处理,计算变换后的均值和方差,作频数直方图、拟合后的正态分布概率密度曲线、Q-Q图。

第二步:对房价进行正态分布化处理;

  1. train["房价"] = np.log1p(train["房价"])#对数变换ln(1+x)
  2. (mu, sigma) = norm.fit(train['房价'])#计算新的均值和方差
  3. print("\n mu = %.2f and sigma = %.2f\n" %(mu,sigma))#打印均值和方差
  4. plt.figure(figsize=(16, 8),dpi=600)#设置图片大小
  5. plt.ylabel('概率密度')#设置纵轴标签
  6. plt.title('房价分布')#设置图片标题
  7. #第一个参数为房价数据,fit=norm代表绘制生态分布密度概率曲线(黑色曲线),kde=True代表绘制核密度估计曲线(蓝色曲线)
  8. #fit_kws,kde_kws,hist_kws分别可以设置黑色曲线,蓝色曲线和直方图的图例标签。
  9. seaborn.distplot(train['房价'] ,fit=norm,kde=True,fit_kws={"label":"正态分布概率密度曲线"},
  10. kde_kws={"label":"核密度估计曲线"},hist_kws={"label":"房价频数"})#绘图
  11. plt.legend()#设置图例
  12. plt.figure(figsize=(16, 8),dpi=600)#设置图片大小
  13. res = stats.probplot(train['房价'], plot=plt)#画出Q-Q图

 

上述结果解释:正态分布化处理后,核密度估计曲线与拟合出的正态分布概率密度曲线更加贴近,Q-Q图中蓝色的点相比处理前更加贴近直线。因此可以认为经过对数变换后的房价数据近似服从正态分布,可以用于接下来的回归任务。

第三步:添加一列总面积并计算偏度;

  1. #增加一个新特征总面积
  2. all_data['总面积'] = all_data['地下室总面积'] + all_data['一楼面积'] + all_data['二楼面积']
  3. from scipy.stats import norm, skew
  4. numeric_feats = all_data.dtypes[all_data.dtypes != "object"].index#找出所有数值型属性的索引值
  5. skewed_feats = all_data[numeric_feats].apply(lambda x: skew(x)).sort_values(ascending=False)#计算所有数值型属性的偏度值
  6. print("\n所有数值型特征的偏度为: \n")#打印
  7. skewness = pd.DataFrame({'偏度值' :skewed_feats})#转换成数据框
  8. skewness.head(10)#查看前十行

第四步:随后对偏度大的列数据进行BOX-COX转换,改善数据的正态性和对称性;

  1. skewness = skewness[abs(skewness) > 0.75]#保留偏度的绝对值大于0.75的特征
  2. print("总共有 %d 个特征需要进行BOX-COX变换" %skewness.shape[0])#打印有多少个特征需要进行变换
  3. from scipy.special import boxcox1p
  4. skewed_features = skewness.index#将特征名称保存在skewed_features中
  5. lam = 0.15#一般情况下0.15为经验值。
  6. for feat in skewed_features:
  7. all_data[feat] = boxcox1p(all_data[feat], lam)#进行BOX-COX转换

第五步:对数据进行独热编码;

使用独热编码来解决类别型数据的离散值问题,方便分析和预测,独热编码的具体原理在此不做详细介绍(就是使用0和1标记数据类别);

  1. all_data = pd.get_dummies(all_data)#热编码
  2. print(all_data.shape)#打印数据形状。

三、使用多个回归预测算法进行房价预测

1. 导入包和数据

  1. from sklearn.pipeline import make_pipeline#级联其他的函数工具
  2. from sklearn.preprocessing import RobustScaler#自动处理离群点的函数
  3. from sklearn.linear_model import ElasticNet, Lasso#弹性回归,套索回归。
  4. from sklearn.kernel_ridge import KernelRidge#核岭回归
  5. from sklearn.ensemble import GradientBoostingRegressor#梯度提升回归
  6. import xgboost as xgb#xgb包中包括极端梯度提升回归模型:XGBRRegressor
  7. #自定义的集成模型类需要寄存sklearn中的三个类:估计器的基类,转换器的混合类,回归估计器的混合类,clone是产生克隆估计器的函数
  8. from sklearn.base import BaseEstimator, TransformerMixin, RegressorMixin, clone
  9. #用于划分数据集,计算模型得分的函数
  10. from sklearn.model_selection import KFold, cross_val_score,train_test_split
  11. train = all_data[:ntrain]#定义训练集
  12. test = all_data[ntrain:]#定义测试集

2. 定义模型

共使用5种回归模型,分别为Lasso、KernelRidge、ElasticNet、GradientBoostingRegressor、XGBRegressor模型。

Lasso模型是在线性回归模型的基础上加入了L1正则化的回归方法;

KernelRidge 带有核函数的岭回归模型,岭回归也是在线性回归模型的基础上加入了12正则化的回归方法;

ElasticNet 是套索回归和岭回归的混合模型,同时引入了L1和L2正则化项;

GradientBoostingRegressor 梯度提升回归模型,是一种袋成回归算法;

XGBRegressor 极端梯度提升回归模型,是在梯度提升回归慕础上经过优化的集成回归算法;

  1. #alpha代表正则化惩罚力度,random_state指随机种子
  2. #正则化项为alpha*//w//_l
  3. Lasso = Lasso(alpha =0.0005, random_state=0)
  4. #alpha代表正则化器罚力度
  5. #正则化项为0.5xalphax//w// 2 2
  6. #kernel=polynomial”指核函数的类型为多项式核函数,K(xxi)=“x*xi)+r)的d次方,degree指poly核中参数d
  7. KRR = KernelRidge(alpha=0.6, kernel='polynomial', degree=2)
  8. #alpha代表正则化惩罚力度,randomstate指随战种子
  9. #正则化项为alpha xll ratio & //w// 1 +05 xalpha a (l -ll ratio) a //w// 2 2
  10. ENet = ElasticNet(alpha=0.0005, l1_ratio=.9, random_state=0)
  11. #nestimators值弱学习器的数目,learningrate值学习率,maxdep场指每个学习器的最大深度:
  12. #max features特征子集中的特征个数,maxfeatures=“sart’代表取特征总数的开方:randomstate指随机种子
  13. GBR = GradientBoostingRegressor(n_estimators=3000, learning_rate=0.05,
  14. max_depth=4, max_features='sqrt', random_state =0)
  15. #nestimators代表指弱学习器的数目:learningrate指学习率:maxdept指树的最大深度
  16. #regalpha指权重的LI正则化项:reglambda指权重12正则化项:
  17. XGBR = xgb.XGBRegressor(n_estimators=2200,learning_rate=0.05, max_depth=3,
  18. reg_alpha=0.4640, reg_lambda=0.8571)

3. 定义训练数据和计算模型均方根误差的方法

  1. class AveragingModels(BaseEstimator, RegressorMixin, TransformerMixin):
  2. def __init__(self, models):
  3. self.models = models
  4. #定义原始模型的克隆来训练数据。
  5. def fit(self, X, y):
  6. self.models_ = [clone(x) for x in self.models]
  7. #训练克隆模型
  8. for model in self.models_:
  9. model.fit(X, y)
  10. return self
  11. #对克隆模型进行预测并求平均
  12. def predict(self, X):
  13. #column_stack将3个n*1的矩阵拼接成n*3的矩阵
  14. predictions = np.column_stack([
  15. model.predict(X) for model in self.models_
  16. ])
  17. return np.mean(predictions, axis=1)#axis=1:压缩列,对各行求均值,返回n*1矩阵。
  18. class StackingAveragedModels(BaseEstimator, RegressorMixin, TransformerMixin):
  19. def __init__(self, base_models, meta_model, n_folds=5):
  20. self.base_models = base_models
  21. self.meta_model = meta_model
  22. self.n_folds = n_folds
  23. #定义原始模型的克隆来训练数据
  24. def fit(self, X, y):
  25. self.base_models_ = [list() for x in self.base_models]
  26. self.meta_model_ = clone(self.meta_model)
  27. kfold = KFold(n_splits=self.n_folds, shuffle=True, random_state=0)
  28. #训练克隆的基础模型(第一阶段的模型),将产生的预测结果存入out_of_fold中,利用out_of_fold中的数据训练元模型(第二阶段的模型)
  29. out_of_fold_predictions = np.zeros((X.shape[0], len(self.base_models)))
  30. for i, model in enumerate(self.base_models):
  31. for train_index, holdout_index in kfold.split(X, y):
  32. instance = clone(model)
  33. self.base_models_[i].append(instance)
  34. instance.fit(X[train_index], y[train_index])
  35. y_pred = instance.predict(X[holdout_index])
  36. out_of_fold_predictions[holdout_index, i] = y_pred
  37. #现在将out_of_fold里的数据作为特征,拿到元模型(第二阶段)的模型中进行拟合
  38. self.meta_model_.fit(out_of_fold_predictions, y)
  39. return self
  40. #用所有的基础模型(第一阶段的模型)预测测试集上的数据,将预测结果的平均值拿到元模型中进行最后的预测,返回的是一个n*l的矩阵
  41. def predict(self, X):
  42. #base_models是一个3*5的二维列表,如下
  43. #每一行代表一个基模型5次交叉验证生成的5个模型,一共有三个基模型,所以有三行
  44. #np.column_stack([model.predict(X) for model in base_models]).mean(axis=1)对每一行5个模型的预测结果求均值
  45. #外层的np.column_stack将三个基模型的预测结果拼接成n*3的矩阵
  46. #因此,meta_features是一个n*3的矩阵,每一列代表一个基模型5次交叉验证的预测的平均值,共有3个基模型所以有三列
  47. meta_features = np.column_stack([
  48. np.column_stack([model.predict(X) for model in base_models]).mean(axis=1)
  49. for base_models in self.base_models_ ])
  50. return self.meta_model_.predict(meta_features)
  51. n_folds = 5#交叉验证的折数
  52. def rmse_cv(model):#编写函数,计算各个模型的均方根误差
  53. kf = KFold(n_folds, shuffle=True, random_state=0).get_n_splits(train.values)#划分交叉验证需要用到的数据集
  54. #scoring="neg_mean_squared_error"是计算均方差回归损失,返回值是负值,故要在函数外添加负号,再开平方,得到均方根误差
  55. rmse= np.sqrt(-cross_val_score(model, train.values, y_train, scoring="neg_mean_squared_error", cv = kf))#计算均方根误差
  56. return(rmse)

4. 通过均方误差评估各模型效果

单独评估各回归模型: 

 综合评估模型组合:

5. 模型应用(预测)

 选用均方误差最小的stacked_averaged_models4进行应用,预测测试集的房价数据:

  1. stacked_averaged_models4.fit(np.array(train),y_train)
  2. y_pred=stacked_averaged_models4.predict(test)
  3. y_pred_real=[]
  4. for i in range(len(y_pred)):
  5. y_pred_real_i=pow(math.exp(1),y_pred[i])-1
  6. y_pred_real.append(y_pred_real_i)
  7. plt.figure(figsize=(20,8),dpi=600)
  8. plt.plot(y_pred_real[0:200],'bo')
  9. plt.xlabel('样本编号')#设置横轴标签
  10. plt.ylabel('预测房价')#设置纵轴标签
  11. plt.show()
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小丑西瓜9/article/detail/626807
推荐阅读
相关标签
  

闽ICP备14008679号