from sklearn import tree
from sklearn.datasets import load_iris
from sklearn2pmml.pipeline import PMMLPipeline
from sklearn2pmml import sklearn2pmml
if __name__ == '__main__':
# 经典数据集
iris = load_iris()
# print(iris.keys())
# 样本特征
x = iris.data
# print(x)
# 分类目标
y = iris.target
# print(y)
# 用决策树分类
pipeline = PMMLPipeline([("classifier", tree.DecisionTreeClassifier())])
# 训练
result = pipeline.fit(x, y)
print(result)
# 输出PMML文件
sklearn2pmml(pipeline, "D:\iris.pmml", with_repr=True)
05-11
872
04-30
670
03-08
1192
07-09
7074
09-28
2544