python - python中的多项式回归模型

Question

据我了解，回归方程可以通过以下函数计算:

import statsmodels.formula.api as smf
fg = smf.ols(formula='X ~ Y', data=data).fit()

我们还可以通过 numpy polyfit 函数进行计算。

numpy.polyfit(x, y, degree)

因为我们可以改变 numpy polyfit 的程度。

在 ols 函数中，我们还可以添加其他自变量，如下所示:

fg = smf.ols(formula='X ~ Y+Y1+Y2', data=data).fit()

所以我的问题是我们可以改变 ols 函数中的拟合顺序/程度吗？或者我们可以在 numpy polyfit 函数中添加另一个自变量吗？

Answer 1

就您提到的 statsmodels 功能而言，公式是使用 patsy 语言指定的(请参阅 http://patsy.readthedocs.io/en/latest/ )。因此，例如，您使用的第一次调用可能如下所示。

fg = smf.ols(formula='X ~ Y + Y**2', data=data).fit()

或

fg = smf.ols(formula='X ~ log(Y)', data=data).fit()