
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)