python - Statsmodels 混合线性模型预测

Question

我正在使用 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()

Answer 1

我知道我迟到了几个月，但如果其他人有同样的问题，最好回答一下。结果对象中提供了所需的参数。它们是 result.fe_params

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

或与结果对象

result.predict(exog=xtest)