【发布时间】:2019-07-23 22:56:11
【问题描述】:
我在 JPMML (scala) 中使用带有许多输入字段的 PMML 模型时遇到问题。在下面找到一个最小示例:加载一个 300x150 像素的图像并将其用作 PCA (python) 的输入:
img = PIL.Image.open(filename)
img = img.resize(STANDARD_SIZE) # 300x150
img = np.array([int(np.mean(a)) for a in img])
pca = PCA(svd_solver=pca_method,n_components = components)
train = pca.fit_transform(train_x)
pipeline = PMMLPipeline(([('pca', pca), ('knn', neigh)]))
sklearn2pmml(pipeline, "/tmp/pca.pmml")
在第二步中,应使用 JPMML (scala) 加载此模型:
val evaluator = new LoadingModelEvaluatorBuilder()
.setLocatable(false)
.load(new File("/tmp/pca.pmml"))
.build()
evaluator.verify()
这将导致非常明显的异常:
Exception in thread "main" org.jpmml.evaluator.InvalidElementException: Model has too many input fields
at org.jpmml.evaluator.ModelEvaluatorBuilder.checkSchema(ModelEvaluatorBuilder.java:135)
at org.jpmml.evaluator.ModelEvaluatorBuilder.build(ModelEvaluatorBuilder.java:115)
...
如果您查看源代码,您可以在ModelEvaluatorBuilder 找到以下限制:
if((inputFields.size() + groupFields.size()) > 1000){
throw new InvalidElementException("Model has too many input fields", miningSchema);
}
所以我的 45k 输入字段太多了。如果我得到正确的 PMML 文档,我只能对 inpt 字段使用原子数据类型(int、char、double 等)。
有什么想法可以实际解决这个限制吗?
【问题讨论】:
标签: numpy machine-learning pca pmml