機械学習モデルをpickleでシリアライズして保存する

学習モデルの作成に無限の時間がかかる。最初は大したことないし毎回学習させとけばいいわ!って思ってたけどいざ実戦に突っ込んだら物凄い時間がかかる。。というわけでモデルを保存しておきたい。Pythonではオブジェクトをこんなふうにシリアライズできるみたいなんで、これで上手く行きました。

import pickle
model_path = "./model.sav"
model = {"a": "b", "c": "d"}
pickle.dump(model, open(model_path, 'wb'))
model2 = pickle.load(open(model_path, 'rb'))
print(model2)

参考

Python scikit-learnで機械学習モデルを保存&ロードする - 株式会社ロカラボ

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です

CAPTCHA