python中实现多项式回归的代码
时间: 2024-10-17 12:06:23 浏览: 24
在Python中,可以使用`numpy`库来创建多项式特征,然后结合`sklearn`库中的`LinearRegression`来进行多项式回归。以下是一个简单的例子:
```python
import numpy as np
from sklearn.linear_model import LinearRegression
from sklearn.preprocessing import PolynomialFeatures
# 假设我们有如下数据
X = np.array([[1], [2], [3], [4], [5]])
y = np.array([2, 4, 6, 8, 10])
# 创建一个二阶多项式特征
poly = PolynomialFeatures(degree=2)
X_poly = poly.fit_transform(X)
# 创建并训练线性回归模型
model = LinearRegression()
model.fit(X_poly, y)
# 预测新值
new_X = np.array([[6], [7]])
new_X_poly = poly.transform(new_X)
predicted_y = model.predict(new_X_poly)
print("预测结果:", predicted_y)
```
在这个例子中,我们将原始的一维数据转换成了二维多项式特征,然后用这个特征向量来训练线性回归模型。注意,`degree`参数决定了特征的最高次数。
相关问题
python拟合多项式回归函数代码
当然可以帮您解决这个问题,以下是Python拟合多项式回归函数代码的示例:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成随机数据
x = np.linspace(-5, 5, 50)
y = x**3 - 5*x**2 + 4*x + 2 + np.random.randn(len(x)) * 10
# 多项式回归
z = np.polyfit(x, y, 3) # 拟合3次多项式回归
p = np.poly1d(z) # 构造回归函数
# 可视化
xp = np.linspace(-5, 5, 100)
plt.plot(x, y, '.', xp, p(xp), '-r')
plt.show()
```
以上代码通过`np.polyfit()`函数实现多项式回归,`np.poly1d()`函数构造回归函数,可视化部分使用了`matplotlib`库。
多项式回归代码python
以下是一个简单的多项式回归的 Python 代码示例,使用 scikit-learn 库:
```python
import numpy as np
from sklearn.linear_model import LinearRegression
from sklearn.preprocessing import PolynomialFeatures
import matplotlib.pyplot as plt
# 生成一些随机数据用于拟合
x = np.array([1, 2, 3, 4, 5, 6])
y = np.array([0.5, 2.5, 2, 4, 3.5, 6])
# 将 x 转为矩阵形式,方便后面的运算
x = x.reshape(-1, 1)
# 将 x 的多项式特征转换为 2 次多项式
poly_features = PolynomialFeatures(degree=2, include_bias=False)
x_poly = poly_features.fit_transform(x)
# 使用线性回归模型拟合多项式特征后的数据
lin_reg = LinearRegression()
lin_reg.fit(x_poly, y)
# 生成一些测试数据用于预测
x_test = np.linspace(0, 7, 100).reshape(-1, 1)
x_test_poly = poly_features.transform(x_test)
# 使用训练好的模型进行预测
y_pred = lin_reg.predict(x_test_poly)
# 绘制拟合曲线和原始数据
plt.plot(x, y, 'o')
plt.plot(x_test, y_pred)
plt.show()
```
在这个例子中,我们生成了一些随机数据用于拟合,然后使用 `PolynomialFeatures` 将输入特征转换为 2 次多项式特征,并使用 `LinearRegression` 对多项式特征进行拟合。最终,我们使用训练好的模型对一些测试数据进行预测,并将拟合曲线和原始数据绘制在图表上。
阅读全文