用给定公式和参数的统计模型计算值

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 statmodels OLS:如何将学习到的模型保存到文件您可以将模型结果保存到pickle文件中，然后从pickle文件中加载模型并使用它进行预测。