线性回归 python代码
时间: 2023-05-20 22:03:25 浏览: 35
线性回归python代码
下面是一个线性回归模型的 Python 代码示例:
```python
import numpy as np
from sklearn.linear_model import LinearRegression
# 训练数据
x = np.array([[1], [2], [3], [4], [5]])
y = np.array([5, 7, 9, 11, 13])
# 建立模型
reg = LinearRegression().fit(x, y)
# 预测
reg.predict(np.array([[6]]))
```
这段代码使用了 NumPy 和 scikit-learn 库。首先,它导入了这两个库,然后使用训练数据训练了一个线性回归模型,最后使用该模型进行预测。
阅读全文