Python进行一元线性回归模型的构建代码和解读
时间: 2024-02-05 10:13:54 浏览: 166
Python实现线型回归模型
一元线性回归是指只有一个自变量和一个因变量之间的线性关系。Python中可以使用scikit-learn库来构建一元线性回归模型。
下面是一元线性回归模型的构建代码:
```python
import numpy as np
from sklearn.linear_model import LinearRegression
# 创建输入数据
X = np.array([1, 2, 3, 4, 5]).reshape((-1, 1))
y = np.array([2, 4, 5, 4, 5])
# 创建模型并拟合数据
model = LinearRegression()
model.fit(X, y)
# 计算R^2分数
r_sq = model.score(X, y)
print('coefficient of determination:', r_sq)
# 预测新的数据点
new_X = np.array([6]).reshape((-1, 1))
new_y = model.predict(new_X)
print('predicted response:', new_y)
```
解读:
1. 首先导入必要的库,包括numpy和scikit-learn的LinearRegression模型。
2. 创建输入数据。这里使用了一个包含5个数据点的简单例子,自变量X是一个列向量,因变量y是一个行向量。
3. 创建模型并拟合数据。使用LinearRegression()创建一个线性回归模型,然后使用fit()方法拟合数据。
4. 计算R^2分数。使用score()方法计算模型的R^2分数,该分数表示模型对因变量的解释程度。该值越接近1,表示模型对数据的拟合越好。
5. 预测新的数据点。使用predict()方法对新的数据点进行预测。
注意,以上代码只是一个简单的例子,实际应用中需要更多的数据和特征工程处理。
阅读全文