python 多项式回归
时间: 2023-10-28 07:57:57 浏览: 93
机器学习__多项式回归.pptx
多项式回归是一种机器学习算法,用于建立一个多项式函数来拟合数据。它适用于非线性数据集,可以通过增加多项式的次数来更好地适应数据。
在Python中,你可以使用scikit-learn库来实现多项式回归。下面是一个示例代码:
```python
import numpy as np
from sklearn.preprocessing import PolynomialFeatures
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error
# 生成一个示例数据集
X = np.array([1, 2, 3, 4, 5]).reshape(-1, 1)
y = np.array([2, 3, 5, 9, 12])
# 创建多项式特征
poly = PolynomialFeatures(degree=2)
X_poly = poly.fit_transform(X)
# 创建线性回归模型
model = LinearRegression()
model.fit(X_poly, y)
# 预测新的数据点
X_test = np.array([6]).reshape(-1, 1)
X_test_poly = poly.transform(X_test)
y_pred = model.predict(X_test_poly)
print("预测值:", y_pred)
```
在上述代码中,我们首先使用`PolynomialFeatures`来创建多项式特征。通过指定`degree`参数来控制多项式的次数。然后,我们使用`LinearRegression`来训练模型并进行预测。
希望这个示例对你有所帮助!如果还有其他问题,请随时提问。
阅读全文