【发布时间】:2017-10-23 09:55:00
【问题描述】:
我需要在 python 中将我的随机森林模型转换为 pmml 格式。我已经从 github 导入了 sklearn2pmml 并尝试创建一个 pmml 文件。我运行下面的代码;
import pandas
import sklearn_pandas
iris = iris.csv
iris_df = pandas.concat((pandas.DataFrame(iris.data[:, :], columns = ["Sepal.Length", "sepal_width", "petal_length", "petal_width"]), pandas.DataFrame(iris.target, columns = ["species"])), axis = 1)
iris_mapper = sklearn_pandas.DataFrameMapper([('sepal_length',None),
('sepal_width', None),
('petal_width', None),
('petal_width', None),
('species',None)])
iris = iris_mapper.fit_transform(iris_df)
from sklearn.ensemble import RandomForestClassifier
iris_X = iris[:, 0:4]
iris_y = iris[:, 4]
iris_classifier = RandomForestClassifier(n_estimators=10)
iris_classifier.fit(iris_X, iris_y)
from sklearn2pmml import sklearn2pmml
sklearn2pmml(iris_classifier, iris_mapper, "randomforest.pmml")
但是,我得到一个错误;
TypeError: The pipeline object is not an instance of PMMLPipeline
任何建议我缺少什么或创建 pmml 格式的其他方法?
【问题讨论】:
标签: python random-forest pmml