给定公式和参数，计算带有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文件加载模型并使用它来预测。

如果您有任何问题，请与我们联系。