使用sklearn框架实现多项式回归。使用框架更方便,可以少写很多代码。
使用一个简单的数据集来模拟,只有几条数据。
代码
如果不用框架,需要自己手动对数据添加高阶项,有了框架就方便多了。sklearn 使用 Pipeline 函数简化这部分预处理过程。
当 PolynomialFeatures 中的degree=1时,效果和使用 LinearRegression 相同,得到的是一个线性模型,degree=2时,是二次方程,如果是单变量的就是抛物线,双变量的就是抛物面。以此类推。
这里有一个 fit_intercept 参数,下面通过一个例子看一下它的作用。
当 fit_intercept 为 True 时,coef_ 中的第一个值为 0