线性回归 python代码
时间: 2023-05-31 07:03:36 浏览: 32
下面是一个线性回归模型的 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 库。首先,它导入了这两个库,然后使用训练数据训练了一个线性回归模型,最后使用该模型进行预测。
相关问题
线性回归 Python代码
下面是一个简单的线性回归 Python 代码示例:
```python
import numpy as np
from sklearn.linear_model import LinearRegression
# 生成数据
np.random.seed(0)
X = np.random.rand(100, 1)
y = 2 + 3 * X + np.random.rand(100, 1)
# 建立模型
model = LinearRegression()
model.fit(X, y)
# 预测
x_new = np.array([[0.5]])
y_new = model.predict(x_new)
# 输出结果
print(y_new)
```
在这个示例中,我们使用 `numpy` 生成了随机数据,然后使用 `LinearRegression` 类从 `sklearn.linear_model` 模块中创建了一个线性回归模型。我们使用 `fit` 方法拟合模型,并使用 `predict` 方法预测新的输入数据。最后,我们使用 `print` 函数输出预测结果。
线性回归python代码实现
以下是一个简单的线性回归 Python 代码实现:
```python
import numpy as np
class LinearRegression:
def __init__(self):
self.w = None
def fit(self, X, y):
X = np.insert(X, 0, 1, axis=1)
self.w = np.linalg.inv(X.T.dot(X)).dot(X.T).dot(y)
def predict(self, X):
X = np.insert(X, 0, 1, axis=1)
return X.dot(self.w)
```
其中,`fit` 方法用于训练模型,`predict` 方法用于预测。`X` 和 `y` 分别是训练数据的特征和标签,`w` 是模型的参数。在 `fit` 方法中,我们首先在 `X` 的第一列插入一列全为 1 的向量,然后使用最小二乘法求解参数 `w`。在 `predict` 方法中,我们同样在 `X` 的第一列插入一列全为 1 的向量,然后使用求解得到的参数 `w` 进行预测。
阅读全文