XGBoostをpmmlに出力

以前作成したモデルを、自社製品に取り込めるようPMML出力機能を追加しました。
https://github.com/HajimeK/machine-learning/blob/master/projects/capstone/capstone.ipynb

以下をimport

from sklearn2pmml import sklearn2pmml
from sklearn2pmml.pipeline import PMMLPipeline

Random Forest Classifierは以下で

pipeline_model = PMMLPipeline([('rf_classifier', rfPredictor)]).fit(X,y)
sklearn2pmml(pipeline_model, './model/rfc.pmml', with_repr = True)

XGBoostは以下で

pmml_model = sklearn2pmml.make_pmml_pipeline(predictor)
sklearn2pmml.sklearn2pmml(pmml_model, './model/xgb.pmml')

無事ファイル出力できました。

ただ、一度ファイルに保存したXGBoostモデルを、ロードしなおしたモデルを変換しようとした際にはうまくいかなかったです。

コメント

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