“TypeError: can't pickle _thread.RLock objects”，同时使用 pickle 保存 Facebook Prophet 模型

Question

I am training a model to predict energy consumption values using Facebook Prophet model. But while saving model using pickle I am getting this error:

TypeError                                 Traceback (most recent call last)
<ipython-input-54-9ef2990419d8> in <module>
      1 with open('forecast_model.pckl', 'wb') as fout:
----> 2     pickle.dump(model, fout)

TypeError: can't pickle _thread.RLock objects

Code :

from fbprophet import Prophet
import pickle
model = Prophet()
model.fit(df)
with open('forecast_model.pkl', 'wb') as fout:   <== Error coming here
    pickle.dump(model, fout)

df.head()

                     ds   y
0   2018-12-19 15:00:02   250.0
1   2018-12-19 16:00:03   295.0
2   2018-12-19 18:00:02   288.0
3   2018-12-19 19:00:02   286.0
4   2018-12-19 20:00:02   229.0

[EDIT] Solution Link : https://github.com/facebook/prophet/issues/1361 :
Code given below worked for me

model = Prophet()
model.stan_backend.logger = None

with open('test.pickle', 'wb') as f:
   pickle.dump(m, f)

Answer 1

I've got the same problem. I reinstalled fbprophet version 0.5, like

pip install -U fbprophet==0.5

if you don't need the latest version:0.6, you can try it.