scala 模型导出问题

2021-01-26 10:35:26 +08:00
 asensio

spark ml 导出模型 pmml 文件不成功,下面是 scala 代码

        val pmml = new PMMLBuilder(trainingData.schema, model).build()
        val targetFile = "target/model.pmml"
        val fis: FileOutputStream = new FileOutputStream(targetFile)
        val fout: StreamResult = new StreamResult(fis)
        JAXBUtil.marshalPMML(pmml, fout)
Exception in thread "main" java.lang.IllegalArgumentException: Expected string, integral, double or boolean data type, got vector data type
	at org.jpmml.sparkml.SparkMLEncoder.createDataField(SparkMLEncoder.java:169)
	at org.jpmml.sparkml.SparkMLEncoder.getFeatures(SparkMLEncoder.java:76)
	at org.jpmml.sparkml.feature.VectorIndexerModelConverter.encodeFeatures(VectorIndexerModelConverter.java:52)
	at org.jpmml.sparkml.FeatureConverter.registerFeatures(FeatureConverter.java:50)
	at org.jpmml.sparkml.PMMLBuilder.build(PMMLBuilder.java:114)
1176 次点击
所在节点    Scala
0 条回复

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/748401

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX