💨

Pythonで Pickleを扱う

に公開
Python
機械学習
pickle
tech

Pickleとは？

変数をバイナリで置いておける仕組み。機械学習の学習後の結果（パラメーター）などをpickle化して置いておく仕組み。ググるとよいページが上の方に来ないのでまとめとして。

①書き込み

pickle.dump(変数, open(ファイルパス, 'wb')) でOK。

import pickle
clf = "Hello, world" # 今は文字列だが何でもOK
path_file = "model.pkl"
pickle.dump(clf, open(path_file, 'wb'))

②読み込み

pickle.dump(open(ファイルパス, 'rb')) でOK。

import pickle
path_file = "model.pkl"
clf = pickle.load(open(path_file, 'rb'))

pandasオブジェクトについて

pandas.DataFrame オブジェクトの場合は pandas.DataFrame.to_pickle() または pandas.read_pickle() が使用できるのでこちらを使用する方がよい。

Discussion