在给定公式和参数的情况下使用 statsmodels 计算值

Question

我已经拟合了一个给定 Poly-3 函数的模型并提取了找到的参数

model = smf.ols(formula='A ~ B + I(B ** 2.0) + I(B ** 3.0)', data=sp)
poly_3 = model.fit()
params = poly_3.params.values

我想保存参数以供以后使用，因为我不想每次都训练模型。参数将例如是[ 0.09525563, 0.09655527, -0.00946222, 0.00056942]

那么，给定公式，参数和一些 x 值如何获得拟合值？ 我在想某事。像这样：

OLS.predict(x=range(20), params=params, formula='A ~ B + I(B ** 2.0) + I(B ** 3.0)')

我明明可以用 Python 自己写公式，但我觉得我不必在这里重新发明轮子！

谢谢！

Answer 1

请参考这个Python statsmodels OLS: how to save learned model to file 您可以将模型结果保存到 pickle 文件中，然后从 pickle 文件中加载模型并使用它进行预测。

如果您有任何问题，请告诉我。