编写一段线性回归代码,要求输出为回归模型参数和模型图像
时间: 2023-06-02 10:03:35 浏览: 101
线性回归代码
好的,以下是一段基于Python的线性回归代码,可以输出回归模型参数和模型图像:
```python
import numpy as np
import matplotlib.pyplot as plt
# 构造数据
x = np.linspace(0, 10, 100)
y = 3 * x + np.random.randn(100)
# 线性回归
X = np.vstack([np.ones(100), x]).T
w = np.linalg.inv(X.T.dot(X)).dot(X.T).dot(y)
# 输出模型参数
print("w0 = {:.2f}, w1 = {:.2f}".format(w[0], w[1]))
# 绘制模型图像
plt.scatter(x, y, alpha=0.6)
plt.plot(x, w[0] + w[1]*x, color='red')
plt.xlabel('x')
plt.ylabel('y')
plt.show()
```
运行以上代码,就可以得到线性回归的模型参数w0和w1,同时也会绘制出回归模型的拟合直线图像。
阅读全文