线性回归(Linear Regression)原理详解及Python代码示例

一、线性回归原理详解

        线性回归是一种基本的统计方法,用于预测因变量(目标变量)与一个或多个自变量(特征变量)之间的线性关系。线性回归模型通过拟合一条直线(在多变量情况下是一条超平面)来最小化预测值与真实值之间的误差。

1. 线性回归模型

        对于单变量线性回归,模型的表达式为:

4736001b906240d9ae7c63ee8a9e102d.png

        其中:

  • y是目标变量。
  • x是特征变量。
  • β0是截距项(偏置)。
  • β1是特征变量的系数。

        对于多变量线性回归,模型的表达式为:

7a34a45745054fe6ba3ed51e68cc001e.png

        其中:

  • y是目标变量。
  • x1,x2,…,xn是多个特征变量。
  • β0是截距项(偏置)。
  • β1,β2,…,βn是各特征变量的系数。

2. 最小二乘法(Ordinary Least Squares, OLS)

        线性回归通过最小二乘法来估计模型参数,即最小化所有预测误差的平方和。对于给定的训练数据集 (xi,yi),目标是找到使得误差平方和最小的 β值。

        误差平方和(损失函数)的公式为:

6fb6e74ddf0841f699893c7315f80cfa.png

        其中 m是样本数量,yi^​ 是第 i个样本的预测值,通过最小化这个损失函数,可以得到最优的模型参数 β。

二、Python代码示例

下面是使用Python实现线性回归的代码示例。我们将使用scikit-learn库来构建和训练线性回归模型,并预测一些数据。

import numpy as np
import matplotlib.pyplot as plt
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error, r2_score

# 生成示例数据
np.random.seed(0)
X = 2 * np.random.rand(100, 1)
y = 4 + 3 * X + np.random.randn(100, 1)

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# 创建线性回归模型并进行训练
lin_reg = LinearRegression()
lin_reg.fit(X_train, y_train)

# 进行预测
y_pred = lin_reg.predict(X_test)

# 打印模型参数
print("截距(Intercept):", lin_reg.intercept_)
print("系数(Coefficients):", lin_reg.coef_)

# 评估模型
mse = mean_squared_error(y_test, y_pred)
r2 = r2_score(y_test, y_pred)
print("均方误差(MSE):", mse)
print("决定系数(R^2):", r2)

# 绘制回归直线
plt.scatter(X_test, y_test, color='blue', label='Actual')
plt.plot(X_test, y_pred, color='red', linewidth=2, label='Predicted')
plt.xlabel('Feature')
plt.ylabel('Target')
plt.title('Linear Regression')
plt.legend()
plt.show()

代码解释:

  1. 数据生成:使用numpy生成随机数据集,特征变量 X 和目标变量 y 满足线性关系并添加一些噪声。
  2. 数据划分:将数据集划分为训练集和测试集,比例为80%训练和20%测试。
  3. 模型训练:使用scikit-learnLinearRegression类创建线性回归模型,并在训练集上进行训练。
  4. 模型预测:使用训练好的模型在测试集上进行预测。
  5. 模型评估:计算均方误差(MSE)和决定系数(R²)来评估模型性能。
  6. 结果可视化:绘制实际值和预测值的散点图以及回归直线。

 

  • 27
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值