在Pandas回归中提供您自己的系数

时间:2014-12-29 15:08:41

标签: python pandas regression

我正在使用Pandas OLS.predict生成模型

mod = smf.ols(formula='response ~ ' + ' + '.join(symbols), data=training)
res = mod.fit()

我后来用于预测:

res.predict(exog=outofsample

有没有一种简单的方法可以将我自己的系数传递给 res 模型?我想将系数存储在JSON文件中,以便在我的脚本中重用它们,而不是每次都生成新的。

1 个答案:

答案 0 :(得分:1)

您可以尝试将res.params中的系数保存/加载到JSON文件中,但最简单的方法是使用本机方法:

res.save('results.pickle')

稍后你可以这样做:

import statsmodels.api as smf
res = smf.load('results.pickle')
res.predict(...)