以前作成したモデルを、自社製品に取り込めるよう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モデルを、ロードしなおしたモデルを変換しようとした際にはうまくいかなかったです。
コメント