我想用Python获取sklearn多项式回归模型的系数,这样我就可以在其他地方编写方程..即ax1^2 + ax + bx2^2 + bx2 + c
我已经在其他地方查看了答案,但似乎无法找到解决方案,除非我只是不知道我在看什么。
from sklearn.preprocessing import PolynomialFeatures
poly_reg = PolynomialFeatures(degree = 2)
X_poly = poly_reg.fit_transform(X_train)
lin_reg_2 = LinearRegression()
lin_reg_2.fit(X_poly,y_train)
lin_reg_2.coef_
Run Code Online (Sandbox Code Playgroud)