Предварительная подготовка к классификации по нескольким меткам - PullRequest
0 голосов
/ 20 декабря 2018

Я должен предварительно обучить модель классификации по нескольким меткам.Я предварительно тренируюсь с набором данных cifar10, и мне интересно, должен ли я использовать для предварительного обучения "categoryor_crossentrpy" (softmax) или "binary_crossentropy" (сигмоид), поскольку в первом случае у меня есть проблема с множественной классификацией

1 Ответ

0 голосов
/ 20 декабря 2018

Вы должны использовать softmax, потому что это дает вам вероятности для каждого класса, независимо от того, сколько их там.Сигмоид, как вы написали, используется с binnary_crossentropy и используется в бинарной классификации (отсюда и бинарная в названии).Надеюсь, теперь стало понятнее.

...