QQ登录

只需要一步,快速开始

 注册地址  找回密码
查看: 2440|回复: 0
打印 上一主题 下一主题

房价预测(线性回归)

[复制链接]
字体大小: 正常 放大

1183

主题

4

听众

2908

积分

该用户从未签到

跳转到指定楼层
1#
发表于 2024-3-31 16:37 |只看该作者 |倒序浏览
|招呼Ta 关注Ta
数据集:使用加利福尼亚房价数据集。+ Y! ?' E! ^+ Y
任务:构建一个模型预测加利福尼亚地区的房价。3 j) i1 j) Y) ~) {4 K, I
挑战:尝试不同的预处理方法(标准化、归一化等)和特征选择技术来改进模型性能。
* E0 z4 {8 {5 N1 p" t: N) m$ o* p线性回归是一种预测数值型数据的经典统计方法,它假设目标值和特征之间存在线性关系。在房价预测任务中,我们可以使用线性回归模型来预测基于多个特征(如房屋大小、位置、年龄等)的房价。以下是使用加利福尼亚房价数据集进行房价预测的示例代码,以及如何应用不同的预处理方法和特征选择技术来改进模型性能。
! T% F3 ?; l  Q% Z
" H$ N* b. i" y. ^  `6 Y. @加载和预处理数据
* Y; G( m0 V, v" E! j首先,我们从scikit-learn中加载加利福尼亚房价数据集,并进行基本的数据预处理。
  1. from sklearn.datasets import fetch_california_housing, l% t) A. ^! [# g1 L0 A

  2. , P) T3 G\\" J, R; `, r4 @: k- A
  3. from sklearn.model_selection import train_test_split  L7 V  i) `; `5 {
  4. 4 N1 K( y4 C3 W3 @
  5. from sklearn.preprocessing import StandardScaler, n. M( ]1 C7 E; e

  6. 6 _; n3 {8 g. X# n7 d% z8 U& Q/ C9 O
  7. from sklearn.linear_model import LinearRegression7 j' K0 w$ C\\" p/ z. E% g
  8. % N) o: Z5 ^+ h5 Q. w( }& c) m( ^/ \
  9. from sklearn.metrics import mean_squared_error
  10. \\" s* |7 y. u1 i+ D9 y% _

  11. 4 F\\" x, e. e, `7 M\\" D$ Y
  12. 2 j+ i$ D& h+ P4 E

  13. $ z- G' w6 {+ W& T
  14. # 加载数据集1 m- r% S* y9 R; h\\" w
  15. ( r, Y; `# p4 Z% W( g* M8 g
  16. housing = fetch_california_housing()7 D3 K\\" P$ n! m

  17. . k' [+ D% W$ y4 g  O) M7 s1 s
  18. X, y = housing.data, housing.target7 E6 W0 p. W( {

  19. + p& Y4 i# f$ I8 a0 c* l4 x& f2 k

  20. + ?/ u  t# E* u; i, e! f7 [

  21. ) L6 X: Q1 ?% W) j8 E* b( T3 O7 F. D% T: P
  22. # 划分训练集和测试集
  23. $ k7 V\\" N6 o0 I8 {0 R
  24. $ \\\" D  c% s' l& g8 d
  25. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)* n  L  e7 Z: C+ ?0 \  ?
  26. + d% n+ Y* \: p9 s. {! {' z$ L

  27. 7 p$ A0 c/ Q( }2 `

  28. ( G5 z5 U7 [# Y, h& e- T6 R) i  x
  29. # 数据预处理:标准化
  30. # c/ W+ f/ w! V
  31. , ]  v6 w. b. P: t
  32. scaler = StandardScaler()
  33.   t. X( R/ H4 u- `2 q; x$ S

  34. ; r  ^+ m9 J9 S+ u6 ]: d; X
  35. X_train_scaled = scaler.fit_transform(X_train)! `, j3 q$ z, t( ^

  36. 1 {. ^3 a2 V( J! H/ G4 {
  37. X_test_scaled = scaler.transform(X_test)
构建和训练线性回归模型
  1. # 创建线性回归模型
    0 X' D7 z  [7 `3 z/ |
  2. model = LinearRegression()
    , r, H5 M- [\" `) \8 M
  3. % K8 c8 Z- D3 x& ?- f8 X
  4. # 训练模型( Q8 s7 Z6 ]  {' [
  5. model.fit(X_train_scaled, y_train)
    8 h' V; O7 O2 F- D4 N* p1 u
  6. ( h& x5 l$ ~) I9 u% X( w1 e3 \: V
  7. # 预测测试集3 z) E4 d# r. G% x+ [, |
  8. y_pred = model.predict(X_test_scaled)
    : f! S4 M6 v  ?5 q+ F/ f

  9. 2 W# a) {# B! G6 p
  10. # 评估模型( u- D. {0 L5 G
  11. mse = mean_squared_error(y_test, y_pred)
    1 W& |6 G6 o2 E5 w
  12. print(f"Mean Squared Error: {mse}")
复制代码
挑战:尝试不同的预处理方法和特征选择, N4 A5 ~+ S( G0 R1 p7 Y- f
预处理方法:除了标准化,你还可以尝试归一化(MinMaxScaler)、对数转换等方法,看看它们如何影响模型的性能。
! N2 {. z9 s! w, b' T特征选择:可以使用不同的特征选择方法(如SelectKBest, SelectFromModel)来选择最有影响力的特征,这有助于模型专注于最重要的信息,提高预测准确性。
  1. from sklearn.feature_selection import SelectKBest, f_regression( q5 \# R( z$ e+ j
  2. % F) M+ j0 Q' f% b
  3. # 特征选择
    4 A* f7 K+ a( d, b1 y) a0 g
  4. selector = SelectKBest(score_func=f_regression, k=5)
    / s* e- T/ d\" n
  5. X_train_selected = selector.fit_transform(X_train_scaled, y_train)
    8 u: Q  V1 f# w6 _& j\" q% g
  6. X_test_selected = selector.transform(X_test_scaled)
    9 I# f* L8 x( v! u! J0 P' k: V0 ^- v

  7. $ g0 W  @* I2 Q
  8. # 使用选择的特征重新训练模型7 _/ ?3 ^/ A) e9 t
  9. model.fit(X_train_selected, y_train); w4 g\" \' R4 k. D/ a* Y
  10. y_pred_selected = model.predict(X_test_selected); c! U8 q\" H- c* x0 e) ~. R! c

  11. $ v% N3 M3 y0 p4 c) `
  12. # 评估- d! q- l% E5 i8 i  ?% T% R
  13. mse_selected = mean_squared_error(y_test, y_pred_selected)! W7 d4 Z: X$ ^# `( N' c
  14. print(f"Mean Squared Error with selected features: {mse_selected}")
复制代码
1 i9 k) E# A- {1 J
zan
转播转播0 分享淘帖0 分享分享0 收藏收藏0 支持支持0 反对反对0 微信微信
您需要登录后才可以回帖 登录 | 注册地址

qq
收缩
  • 电话咨询

  • 04714969085
fastpost

关于我们| 联系我们| 诚征英才| 对外合作| 产品服务| QQ

手机版|Archiver| |繁體中文 手机客户端  

蒙公网安备 15010502000194号

Powered by Discuz! X2.5   © 2001-2013 数学建模网-数学中国 ( 蒙ICP备14002410号-3 蒙BBS备-0002号 )     论坛法律顾问:王兆丰

GMT+8, 2025-12-7 13:11 , Processed in 0.445587 second(s), 50 queries .

回顶部