【发布时间】:2018-10-23 08:30:58
【问题描述】:
我想在 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_
【问题讨论】:
-
我有,但我没有使用 make_pipeline 库,所以我没有 .steps 方法可以使用...
标签: python scikit-learn