求解线性模型参数的均方误差python代码
时间: 2024-05-10 16:14:54 浏览: 82
机器学习领域 线性回归模型数理推导及python代码实现
以下是求解线性模型参数的均方误差的Python代码示例:
```python
import numpy as np
# 生成随机数据
X = np.random.rand(100, 5)
y = np.dot(X, np.array([1, 2, 3, 4, 5])) + np.random.randn(100)
# 添加偏置项
X = np.c_[np.ones((X.shape[0])), X]
# 计算参数
theta = np.dot(np.linalg.inv(np.dot(X.T, X)), np.dot(X.T, y))
# 预测值
y_pred = np.dot(X, theta)
# 计算均方误差
mse = np.mean((y_pred - y) ** 2)
print('模型参数:', theta)
print('均方误差:', mse)
```
其中,`X`是自变量数据,`y`是因变量数据。`np.dot`表示矩阵乘法,`np.linalg.inv`表示求矩阵的逆矩阵,`np.mean`表示求均值。
阅读全文