Я построил нейронную сеть для мультиклассовой классификации в Keras и мне нужно экспортировать ее в файл PMML.Я знаю о пакете keras2pmml, но, к сожалению, он поддерживает только проблемы двоичной классификации.
Мой код:
modelnew2 = Sequential()
# Add the first layer
modelnew2.add(Dense(output_dim=9, activation='tanh',input_shape=(12,)))
modelnew2.add(Dense(output_dim=8, input_dim=9, activation='tanh'))
# Add the output layer
modelnew2.add(Dense(5, activation='softmax'))
# Compile the model !!!! Sinds je met declining learning rate werkt, ga je wel vooruit. Hierop zou je verder kunnen
#met hyperparameter tuning
#sgd = SGD(lr=0.01)
learning_rate = 0.2
epochs=20
decay_rate = learning_rate / epochs
Adam(lr=learning_rate, beta_1=0.9, beta_2=0.999, epsilon=None, decay=decay_rate, amsgrad=False)
modelnew2.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
При запуске keras2pmml вы получаете Keyerror: 'softmax', поскольку он не поддерживает преобразование модели классификации мультикласса в pmml.
Как мне экспортировать мою модель Keras в pmml?