数学建模社区-数学中国

标题: 岭回归 [打印本页]
作者: 2744557306    时间: 2024-3-15 17:22
标题: 岭回归
代码执行了以下操作：
7 c& Y" o2 i; |- }9 Z7 l
* m' D5 `0 [$ p! M6 U1.导入所需库：
  1.    import numpy as np, Q, o2 Q) a9 ]+ k

  2. 9 Q9 {- {* w0 t4 w8 Y
  3.    import pandas as pd
    - S7 j1 z' N& j& y

  4. " J( l' e, O, p& I& y
  5.    import matplotlib.pyplot as plt
    9 ~! t: U- H# k+ F

  6. 8 b- W: O0 E) ^1 h% {9 ?
  7.    from sklearn.linear_model import Ridge, RidgeCV
复制代码
2.定义源数据：
  1.    df = pd.DataFrame({2 g$ e* P1 o- I' K
  2. & T. x1 F1 f# ?  }8 h4 y
  3.        'x1': [7, 1, 11, 11, 7, 11, 3],
    ; W8 M8 ~* T- ]* b8 q
  4. 1 y, W7 v! Y; p% |8 i3 W/ `- M
  5.        'x2': [26, 29, 56, 31, 52, 55, 71],
    : E, r, Z$ I% V4 w, ^

  6. ) ?# C: z/ `6 i5 w4 U# O1 Q
  7.        'y': [78.5, 74.3, 104.3, 87.6, 95.9, 109.2, 102.7],
    7 q1 e, @* y9 x( |" w$ b! n
  8. $ O. w/ n. y$ n
  9.    })
复制代码
创建了一个包含两个自变量（x1 和 x2）和一个因变量（y）的 DataFrame。
0 g( M7 K- l6 g
0 W0 H* O- u5 k. c$ s% S3.将数据转换为数组格式：
  1.    X = np.array(df[['x1', 'x2']])  # 特征集
    & ]5 V4 w% N0 C: V6 j
  2. ) Q8 f( x1 ^  L! B& C
  3.    y = np.array(df[['y']])          # 目标集
复制代码
4.遍历不同的正则化参数 ( k )：
  1.    k_array = np.logspace(-4, 1.5, 100)! g( x1 [+ L) J9 ^  d- I

  2. ! |" o0 n' Z/ r9 L
  3.    x1_list, x2_list = [], []1 A. i3 W. E, V: R9 Z1 [' h) _# Y
  4. + P: |3 e  ]2 a0 M1 ~3 F3 o
  5.    for k in k_array:7 `2 b" h$ a6 h0 u8 Y+ o
  6. # l! q! i/ A. A8 ^) Z5 O
  7.        model = Ridge(alpha=k).fit(X, y)
    4 E2 c! z! S, F% J) L4 M

  8. - O! K/ N7 i+ U# u+ W; |
  9.        x1_list.append(model.coef_[0][0])7 ^3 K$ p6 c3 P) _
  10. # [! H6 t- {1 e: Y! [$ O$ ~
  11.        x2_list.append(model.coef_[0][1])
复制代码
遍历了一系列正则化参数 ( k )，对每个参数值使用 Ridge 回归模型拟合数据，并将得到的系数存储到 x1_list 和 x2_list 中。% W7 K/ j4 y& H

) C5 }6 A' v4 p" t$ `8 y/ B% i" B( Y) ?5.绘制岭迹图：
  1.    plt.scatter(k_array, x1_list)3 P) v7 n" R' S' ?
  2. ( T3 u* X5 y; R/ Y
  3.    plt.scatter(k_array, x2_list)
    ! E8 d; a( r7 {4 g

  4. 4 h  P6 P. ?9 u" X( g$ v
  5.    plt.plot(k_array, x1_list, label='x1')# L5 q% @) n; _

  6. . p( F/ l& T& K% R
  7.    plt.plot(k_array, x2_list, label='x2')# t: d8 p) T" a9 E; t
  8. 7 F8 L; T1 t9 _' ^: _8 C
  9.    plt.legend()
复制代码
绘制了岭迹图，展示了不同正则化参数 ( k ) 下，系数 ( b1 ) 和 ( b2 ) 的变化趋势。# Z+ t% a: b6 [

0 i: P) s+ ?1 V( R6 Z' a" ]/ G6.自动匹配最佳 ( k ) 值：
  1.    model2 = RidgeCV().fit(X, y)
复制代码
使用 RidgeCV 自动匹配最佳正则化参数 ( k ) 并拟合数据。, C+ D  B) Q3 M; n
+ Q- O6 n8 l9 ?: E
7.输出最佳模型的参数和拟合优度：
  1.    b0 = model2.intercept_[0]  # 截距* N- K# |; ~/ `1 J8 B  W  }
  2. 9 u! Y" {1 c: \% A
  3.    b1, b2 = model2.coef_[0][0], model2.coef_[0][1]  # 系数
    ! j8 \; Y2 I8 p7 h& E$ k
  4. & z! m+ d$ Q, c  d% b' v, p4 g9 J' s
  5.    print('y = {:.4f} + {:.4f}*x1 + {:.4f}*x2'.format(b0, b1, b2))# U& W1 b$ {- m. j3 M* Q" Q
  6. " N. V7 m' J1 T  @7 _
  7.    print('R_square =', model2.score(X, y))% u- X5 _+ r1 h/ F

  8. + H5 f9 r% o9 g* w6 M0 z
  9.    print('k =', model2.alpha_)
复制代码
输出了最佳模型的截距、系数、拟合优度以及对应的最佳正则化参数 ( k )。
' ~5 H- R4 _9 i  Y$ v; C! `通过这段代码，你可以对 Ridge 回归模型进行正则化，并选择最佳的正则化参数来提高模型的泛化能力。同时，岭迹图可以帮助你直观地观察不同正则化参数下模型系数的变化趋势。
. T+ W$ ?; V7 m/ T6 Q2 [4 C
9 [" _: M; f  x/ i' f! y" I- C) k; T( @. J2 p
" }5 [; ~8 p3 ~8 {: n

) Y+ ^( H$ d( [, {9 ]
' u  i7 N* m. d# [  ~

12.ridge_regression.py

1.17 KB, 下载次数: 0, 下载积分: 体力 -2 点

售价: 2 点体力  [记录]  [购买]





欢迎光临 数学建模社区-数学中国 (http://www.madio.net/) Powered by Discuz! X2.5