回归预测的算法代码是什么
时间: 2023-09-08 11:17:18 浏览: 153
Regression回归算法代码
回归预测的算法有很多种,比如线性回归、多项式回归、岭回归等。下面给出一个简单的线性回归算法的Python代码示例:
```python
import numpy as np
class LinearRegression:
def __init__(self):
self.w = None
def fit(self, X, y):
X = np.hstack((np.ones((X.shape[0], 1)), X)) # 添加一列全为1的截距项
self.w = np.linalg.inv(X.T @ X) @ X.T @ y
def predict(self, X):
X = np.hstack((np.ones((X.shape[0], 1)), X))
return X @ self.w
```
其中,fit方法用于训练模型,传入训练数据X和标签y,通过最小二乘法计算出权重向量w;predict方法用于预测,传入测试数据X,返回预测结果。
阅读全文