python 实现线性回归
时间: 2024-06-06 21:04:09 浏览: 133
Python实现线性回归可以使用scikit-learn库,它提供了一个简单的线性回归类LinearRegression。下面是一个简单的示例:
```python
# 导入相关库
import numpy as np
from sklearn.linear_model import LinearRegression
# 创建样本数据
X = np.array([, , , , ])
y = np.array([2, 4, 5, 4, 5])
# 创建线性回归模型
model = LinearRegression()
# 训练模型
model.fit(X, y)
# 预测值
y_pred = model.predict(X)
# 输出系数和截距
print("Coefficients: ", model.coef_)
print("Intercept: ", model.intercept_)
# 输出预测值
print("Predicted values: ", y_pred)
```
阅读全文