Экспорт модели нейронной сети в pmml - PullRequest
0 голосов
/ 18 сентября 2018

Я пытаюсь извлечь PMML из модели, созданной с помощью пакета neuralnet, но у меня есть ошибка.Я полагаю, это связано с тем, что целевой переменной является «ХОРОШО + ПЛОХО», потому что, когда я пытаюсь использовать только одну из них, она работает нормально.Тем не менее, моя цель является фактором, поэтому я не знаю, как я могу решить эту проблему.

Спасибо!

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")

Это ошибка, которую я получаю:

SEVERE: не удалось преобразовать 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.encodejMM39 (конвертер)) в org.jpmml.rexp.Main.run (Main.java:149) в org.jpmml.rexp.Main.main (Main.java:97) Ошибка в .convert (временный файл, файл, конвертер, converter_c)Ласспат, многословный): 1

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...