V2EX = way to explore
V2EX 是一个关于分享和探索的地方
Sign Up Now
For Existing Member  Sign In
asensio
V2EX  ›  Scala

scala 模型导出问题

  •  
  •   asensio · Jan 26, 2021 · 4220 views
    This topic created in 1919 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   ·   3448 Online   Highest 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 40ms · UTC 10:49 · PVG 18:49 · LAX 03:49 · JFK 06:49
    ♥ Do have faith in what you're doing.