💨
Pythonで Pickleを扱う
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