PolynomialFeatures使用说明
Sklearn非线性回归跟实现线性回归方法类似,也是用LinearRegression模型,但是在训练之前要用PolynomialFeatures来进行特征的构造。
我们先来看PolynomialFeatures的作用。
from sklearn.preprocessing import PolynomialFeatures
x = [[3], [4]]
poly_reg = PolynomialFeatures(degree=2)
x_poly = poly_reg.fit_transform(x)
print x_poly
执行结果