用预测的模型预测训练集,不准,感觉模型没有训练
import pandas
from sklearn.datasets import load_iris
iris_df = load_iris()
iris_X = iris_df.data
iris_y = iris_df.target
from sklearn.tree import DecisionTreeClassifier
from sklearn2pmml.pipeline import PMMLPipeline
pipeline = PMMLPipeline([
("classifier", DecisionTreeClassifier())
])
pipeline.fit(iris_X, iris_y)
from sklearn2pmml import sklearn2pmml
sklearn2pmml(pipeline, "DecisionTreeIris1.pmml", with_repr = True)
from pypmml import Model
model = Model.load("DecisionTreeIris1.pmml")
model.predict([5.1,3.5,1.4,0.2])