将神经网络模型导出到pmml

时间:2018-09-18 14:08:18

标签: java r neural-network pmml

我正在尝试从使用Neuronet软件包创建的模型中提取PMML,但是出现错误。 我相信这是由于目标变量为“ GOOD + BAD”,因为当我仅尝试其中之一时,它运行正常。但是,我的目标是一个因素,所以我不知道如何解决此问题。

谢谢!

dfGoodBad <- data.frame(V1=c(.2,.1,.2,.2,.4,.3),
                            V2=c(.9,.1,.4,.5,.5,.8),

TARGET<-factor(c('GOOD','BAD','BAD','BAD','GOOD','GOOD')))

dfGoodBad$GOOD<-ifelse(dfGoodBad$TARGET=='GOOD',1,0)
dfGoodBad$BAD<-ifelse(dfGoodBad$TARGET=='BAD',1,0)

nn <- neuralnet(GOOD + BAD ~ V1+V2, data = dfGoodBad, algorithm = "rprop+", linear.output = F, hidden = 3)

library(r2pmml)
r2pmml(nn, "nn.pmml")

这是我得到的错误:

  

严重:无法在以下位置转换java.lang.IllegalStateException   org.jpmml.rexp.RVector.asScalar(RVector.java:56)在   org.jpmml.rexp.NNConverter.encodeSchema(NNConverter.java:58)在   org.jpmml.rexp.ModelConverter.encodePMML(ModelConverter.java:69)在   org.jpmml.rexp.Converter.encodePMML(Converter.java:39)在   org.jpmml.rexp.Main.run(Main.java:149)在   org.jpmml.rexp.Main.main(Main.java:97)

     

线程“ main”中的异常java.lang.IllegalStateException在   org.jpmml.rexp.RVector.asScalar(RVector.java:56)在   org.jpmml.rexp.NNConverter.encodeSchema(NNConverter.java:58)在   org.jpmml.rexp.ModelConverter.encodePMML(ModelConverter.java:69)在   org.jpmml.rexp.Converter.encodePMML(Converter.java:39)在   org.jpmml.rexp.Main.run(Main.java:149)在   org.jpmml.rexp.Main.main(Main.java:97).convert(tempfile,   文件,转换器,converter_classpath,详细):1

0 个答案:

没有答案