python - 是否有用于为 'patsy' 指定 "baseline"模型的 'statsmodels' 公式语法

Question

我想使用公式为一些使用 statsmodels 拟合的模型指定“基线”模型 例如，我希望能够指定一个公式以传递给 olm 或 Logit 模型，它简单地预测所有观测值的观测因变量的平均值。我知道我可以简单地通过计算因变量的观测值的平均值来获得这些数字，但我想要一个产生这些结果的模型(例如，这样我就可以使用它的方法)。是否有一个 patsy 语法来完成这个？

Answer 1

如果您使用的公式只有截距项，那么您将得到因变量的均值/平均值:

import statsmodels.formula.api as smf

data={'y': [1,5,9],                       # mean(y) == 5 
      'X': [[2013], [0.001], [19.99]]     # doesn't matter
      }
model = smf.ols('y ~ 1', data=data).fit()
model.predict(3.14)                       # ==> 5