基于sciket-learn实现多项式回归

本文介绍了如何利用scikit-learn库进行多项式回归。通过一个二次曲线的例子,展示了线性回归在处理非线性问题时的不足,并通过增加特征维度,使用多项式回归来提高拟合效果。通过调整PolynomialFeatures的degree参数,可以观察不同阶数多项式对拟合的影响。
摘要由CSDN通过智能技术生成

多项式回归在思想上和线性回归是一致的,都使用一条线去拟合样本值,进入用得出的模型去进行预测,在样本特征呈现出线性特性时,我们可以用线性回归去做预测,但是在样本特征很复杂的时候,线性回归往往会呈现出欠拟合的状态,这时就需要多项式回归。

先来看一个小例子,给定一条二次曲线y=2x^2 + 2x,生成带噪声的100个样本点,绘制出图像 ,是我们熟悉的二次方程。

x = np.random.uniform(-3, 3, size = 100)
X = x.reshape(-1, 1)

y = 2 * x**2 + x + 2 + np.random.normal(0, 1, size = 100)

plt.scatter(x, y)
plt.show()

然后我们用线性回归来拟合这条曲线,看看会出现什么情况

from sklearn.linear_model import LinearRegression

lin_reg = LinearRegression()
lin_reg.fit(X, y)

y_predict = lin_reg.predict(X)

plt.scatter(x, y)
plt.plot(x,
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值