【Scikit-learn】MinMaxScaler、StandardScaler等を保存する

正規化で使ったMinMaxScalerやStandardScalerのオブジェクトを保存する方法を紹介します。

結論としては、基本的にはpickleを使うだけです。なのでMinMaxScaler以外の物でも保存できます。pickleについては参考文献を見てもらえればと思います。

まぁ早速コードを見てみましょ!

保存

まず保存するコードを紹介します。

from pickle import dump
from sklearn.preprocessing import MinMaxScaler
scaler = MinMaxScaler()
dump(scaler, open("sample.pkl", "wb"))

まぁdumpを使って保存しているだけですね。

復元

次に保存したMiMaxScaler()インスタンスを復元してみます。

from pickle import load
scaler = load(open("sample.pkl", "rb"))

参考文献

https://docs.python.org/ja/3/library/pickle.html

タイトルとURLをコピーしました