Как определить функцию потерь для мульти-выходной-мультиклассовой классификации в Керасе - PullRequest
Новая
       90

Как определить функцию потерь для мульти-выходной-мультиклассовой классификации в Керасе

0 голосов
/ 27 сентября 2019

Я решаю задачу классификации с несколькими выходами.Всего в задаче 10 классов.Моя нейронная сеть имеет 81 выходной узел.Все 81 значения представляют некоторый класс из 10 классов.

В этом сценарии как я могу определить свою функцию потерь?Я использую Keras Framework.

Ниже приведен код для моей нейронной сети. 

model = keras.models.Sequential()

model.add(Conv2D(32, kernel_size=3, padding='same', input_shape=(9,9,1)))
model.add(Activation('relu'))
model.add(Conv2D(16, kernel_size=3, padding='same'))
model.add(Activation('relu'))
model.add(Conv2D(1, kernel_size=3, padding='same'))
model.add(Flatten())
model.add(Dense(81))
...