是否有'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