Statsmodels混合线性模型预测

时间:2018-05-11 21:02:49

标签: python python-3.x statsmodels mixed-models panel-data

我正在使用Python中的statsmodels MixedLM包来估计混合线性模型。在拟合模型之后,我现在想做出预测,但我正在努力理解“预测”。方法。

statsmodels文档(http://www.statsmodels.org/dev/generated/statsmodels.regression.mixed_linear_model.MixedLM.predict.html)表明predict方法接受一个包含已估计模型参数的数组。我该如何检索这个数组?

y = raw_data['dependent_var']
X = raw_data[['var1', 'var2', 'var3']]
groups = raw_data['person_id']

model = sm.MixedLM(endog=y, exog=X, groups=groups)
result = model.fit()

1 个答案:

答案 0 :(得分:1)

我知道我要迟到几个月,但是如果有人遇到同样的问题,可以回答。所需的参数在结果对象中可用。它们是result.fe_params

model.predict(reresult.fe_params, exog=xest)

或带有结果对象

result.predict(exog=xtest)