塩焼きブログ

塩焼きに関しての研究内容を公開しています

機械学習モデルを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で機械学習モデルを保存&ロードする - 株式会社ロカラボ