有没有' patsy'用于指定" baseline"的公式语法用于' statsmodels的模型'

时间:2014-03-22 00:11:44

标签: python linear-regression statsmodels

我想使用公式来指定"基线"使用statsmodels拟合某些模型的模型例如,我希望能够指定一个公式来传递给olmLogit模型,该模型只是简单地预测观察所有观察的因变量。我知道我可以通过计算因变量的观察值的平均值来获得这些数字,但我希望有一个产生这些结果的模型(例如我可以使用它的方法)。是否有patsy语法来完成此操作?

1 个答案:

答案 0 :(得分:2)

如果您使用仅包含截距项的公式,那么您将获得因变量的平均值/平均值:

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