小编Sim*_*ith的帖子

在 Python 中获取 sklearn 多项式回归模型的系数

我想用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)

python scikit-learn

5
推荐指数
1
解决办法
1万
查看次数

标签 统计

python ×1

scikit-learn ×1