python, R, vimでデータマイニング

python, R, vim で疑問に思ったことなどを

LGBMClassifierのモデルをdumpしたい

ちょっと調べてみたところ
from sklearn.externals import joblib
で行けるらしい。
試してみる。
→ 問題なくできました。

1. モデルの作成

Code:
# %%
import pandas as pd
from lightgbm import LGBMClassifier

from utils4ml.utils import create_X_y_of_bank
from utils4ml.sklearnwrappers import train_test_split_roc_auc_score

X, y = create_X_y_of_bank()
score, estimator = train_test_split_roc_auc_score(
    LGBMClassifier(silent=True,),
    X,
    y.y,
)
# %%

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)
# %%