pmml模型文件保存和读取

#模型保存pmml
from sklearn2pmml import sklearn2pmml,PMMLPipeline
pipeline = PMMLPipeline([('classifier',xgb_model)])
pipeline.fit(train_xgb.drop('target',axis=1),train_xgb['target'])
sklearn2pmml(pipeline, '模型文件/{}.pmml'.format('xgb模型20220119'),with_repr = True,debug = True)

# 读取pmml模型文件
from pypmml import Model
model = Model.fromFile('模型文件/xgb模型20220119.pmml')

 

posted @ 2022-03-31 10:36  地豆苗  阅读(1278)  评论(0)    收藏  举报