机器学习预测模型导出pmml文件

from sklearn.datasets import load_iris
from sklearn.ensemble import RandomForestClassifier
from sklearn2pmml import PMMLPipeline, sklearn2pmml
import pandas as pd
from pandas import DataFrame

iris = load_iris()
#创建带有特征名称的
iris_df = pd.DataFrame(iris.data, columns=iris.feature_names)
#创建模型管道
iris_pipeline = PMMLPipeline([("classifier", RandomForestClassifier())])
#训练模型
iris_pipeline.fit(iris_df, iris.target)
#导出模型
sklearn2pmml(iris_pipeline, "test_Iris.pmml")