项目场景:
sklearn机器学习建模后,为了实现模型跨平台工程化调用功能,考虑将训练好的模型转化为PMML文件,便于JAVA后台调用
实战:
1.简单示例:
from sklearn2pmml.pipeline import PMMLPipeline
from sklearn2pmml import sklearn2pmml
from sklearn.tree import DecisionTreeClassifier
import pandas as pd
from sklearn.datasets import load_iris
# 读取数据
data = load_iris()
x = data.data
y = data.target
# 通过管道转换
pipe = PMMLPipeline([('clf',DecisionTreeClassifier())])
pipe.fit(x,y