我正在使用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()
答案 0 :(得分:1)
我知道我要迟到几个月,但是如果有人遇到同样的问题,可以回答。所需的参数在结果对象中可用。它们是result.fe_params
model.predict(reresult.fe_params, exog=xest)
或带有结果对象
result.predict(exog=xtest)