python 线性回归 代码
时间: 2023-10-12 16:55:59 浏览: 130
线性回归python代码
当然,我可以帮你提供一段 Python 的线性回归代码示例。以下是一个简单的例子:
```python
import numpy as np
from sklearn.linear_model import LinearRegression
# 训练数据
X = np.array([[1], [2], [3], [4], [5]])
y = np.array([2, 4, 6, 8, 10])
# 创建线性回归模型
model = LinearRegression()
# 训练模型
model.fit(X, y)
# 预测
x_test = np.array([[6], [7]])
y_pred = model.predict(x_test)
print("预测结果:", y_pred)
```
在这个例子中,我们使用了 `numpy` 来创建了输入数据 `X` 和目标变量 `y`。然后,我们使用 `sklearn` 库中的 `LinearRegression` 类来创建线性回归模型。我们将数据传递给 `fit()` 方法来训练模型,并使用 `predict()` 方法来进行预测。
希望这个例子对你有帮助!如果你有其他问题,请随时提问。
阅读全文