使用LinearRegression怎样得出线性回归线
时间: 2024-05-12 13:19:51 浏览: 73
基于python实现线性回归LinearRegression
使用LinearRegression得出线性回归线的过程如下:
1. 导入LinearRegression模型
```python
from sklearn.linear_model import LinearRegression
```
2. 创建LinearRegression模型
```python
reg = LinearRegression()
```
3. 使用fit方法拟合数据
```python
reg.fit(X, y)
```
其中,X为特征矩阵,y为目标变量。
4. 得出模型的截距和系数
```python
intercept = reg.intercept_
coef = reg.coef_
```
5. 绘制线性回归线
假设有两个特征变量x1和x2,线性回归方程为:
y = intercept + coef1 * x1 + coef2 * x2
可以选择其中一个特征变量作为x轴,另一个变量作为y轴,绘制散点图。然后,使用上述方程计算每个点的预测值,将预测值与对应的特征变量值连接起来,就得到了线性回归线。
阅读全文