【Python】時系列予測ライブラリProphetで学習したモデルを保存・呼び出しする
こんにちは。
随分前ですが、Facebookの時系列予測ライブラリのProphetについて記事を書きました。
ちょっぴり本格的に使う機運が高まってきて、「そういえばどうやってモデルを保存・呼び出しするんだろう」って思って、調べてみました。
今回はその調べた内容についてまとめます。
やること
以下の二つをやってみようと思います。
- Prophetを使って学習したモデルを、pickle形式で保存
- pickle形式で保存された学習済みモデルを読み込み
モデルの保存
思ったより普通にできます笑。以下のように、pickleとして保存するだけですね。
# modelの保存 pkl_model_path = 'path/to/prophet-model.pkl' with open(pkl_model_path, 'wb') as f: pickle.dump(m, f)
モデルの読み込み
読み込みも、同じ要領でできます。
with open('path/to/model.pkl', 'rb') as f: m = pickle.load(f)
こんな感じで。
