asensio
V2EX  ›  Scala

scala 模型导出问题

  •  
  •   asensio · Jan 26, 2021 · 4589 views
    This topic created in 1932 days ago, the information mentioned may be changed or developed.

    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)
    
    No Comments Yet
    About   ·   Help   ·   Advertise   ·   Blog   ·   API   ·   FAQ   ·   Solana   ·   930 Online   Highest 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 40ms · UTC 21:20 · PVG 05:20 · LAX 14:20 · JFK 17:20
    ♥ Do have faith in what you're doing.