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