LGBMClassifierのモデルをdumpしたい
LGBMClassifierのモデルをdumpしたい
Table of Contents
ちょっと調べてみたところ
from sklearn.externals import joblib
で行けるらしい。
試してみる。
→ 問題なくできました。
1. モデルの作成
Code:
2. モデルのdump
Code:
# %% from sklearn.externals import joblib joblib.dump(estimator, 'estimator.pkl') # %%
3. モデルのload
Code:
# %% clf = joblib.load('estimator.pkl') # %%
4. loadしたモデルでpredict
Code:
# %% pred = clf.predict_proba(X) # %%