线性回归python代码实现
时间: 2023-05-20 09:03:28 浏览: 119
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` 进行预测。
阅读全文