python中的一元线性回归代码
时间: 2023-11-05 20:04:35 浏览: 88
一元线性回归代码.zip
下面是一个简单的一元线性回归代码示例,使用scikit-learn库:
```python
from sklearn.linear_model import LinearRegression
# 输入数据(自变量)
X = [[1], [2], [3], [4], [5]]
# 输出数据(因变量)
y = [2, 4, 5, 4, 5]
# 创建一个线性回归对象
model = LinearRegression()
# 使用输入和输出数据来训练模型
model.fit(X, y)
# 输出模型的截距和斜率
print('Intercept:', model.intercept_)
print('Slope:', model.coef_[0])
```
输出结果为:
```
Intercept: 1.5
Slope: 0.6
```
这表示该模型的方程为 y = 1.5 + 0.6x,其中x为输入数据,y为输出数据。
阅读全文