当前位置:   article > 正文

基于多元线性回归的波士顿房价预测_采用波士顿房价预测数据集进行多元线性回归任务分析

采用波士顿房价预测数据集进行多元线性回归任务分析

 本项目利用boston房价数据集联系简单的线性回归,若预测效果不够理想,可进一步进行非线性回归尝试。

  1. # -*- coding: utf-8 -*-
  2. from sklearn import datasets
  3. import matplotlib.pyplot as plt
  4. import numpy as np
  5. import pandas as pd
  6. from sklearn import linear_model
  7. import warnings
  8. warnings.filterwarnings('ignore')
  9. boston=datasets.load_boston()
  10. x=boston.data
  11. y=boston.target
  12. print(x.shape)
  13. print(boston.DESCR)
  14. '''
  15. #练习
  16. clf = linear_model.LinearRegression()
  17. x=np.array([2,3,5,7,6]).reshape(-1,1)
  18. y=np.array([6,10,14.5,21,18.5])
  19. print(plt.scatter(x,y,color='blue'))
  20. clf.fit(x,y) #训练模型
  21. b,a=clf.coef_, clf.intercept_
  22. print(b,a)
  23. x=[[4]]
  24. print(clf.predict(x))
  25. print(plt.plot(x, a+b*x, color = 'red'))
  26. '''
  27. #波士顿房价
  28. x=pd.DataFrame(boston.data, columns=boston.feature_names)
  29. y=pd.DataFrame(boston.target,columns=['MEDV'])
  30. print(plt.scatter(x['RM'],y,color='blue'))
  31. print(plt.scatter(x['LSTAT'],y,color='blue'))
  32. import statsmodels.api as sm
  33. #statsmodels中线性回归模型没有截距项,下行给训练集加上一列数值为1的特征
  34. x_add1=sm.add_constant(x)
  35. model=sm.OLS(y,x_add1).fit()
  36. print(model.summary())
  37. '''
  38. #输出结果
  39. coef std err t P>|t| [0.025 0.975]
  40. ------------------------------------------------------------------------------
  41. const 36.4595 5.103 7.144 0.000 26.432 46.487
  42. CRIM -0.1080 0.033 -3.287 0.001 -0.173 -0.043
  43. ZN 0.0464 0.014
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/代码探险家/article/detail/842044
推荐阅读
相关标签
  

闽ICP备14008679号