Keras LSTM prevendo apenas 1 categoria, na classificação multi-categoria - como corrigir?

votos
1

Eu tenho um conjunto de dados de texto que tem igual número de etiquetas - 0,1,2,3,4. Corri o Keras binary classificationexemplo LSTM(exemplo title) em seu site com o meu conjunto de dados ea linha de compilação alterado para

model.compile(loss='categorical_crossentropy', optimizer='adam', class_mode=categorical)

Mas o modelo prevê apenas 1 categoria, ou seja, 20%a precisão de forma consistente.

Poderia, por favor me ajudar a corrigi-lo / alterar as configurações conforme necessário?

Publicado 29/05/2015 em 03:53
fonte usuário
Em outras línguas...                            


1 respostas

votos
6

Você precisará modificar o com

model.add(Dense(nb_classes))

onde nb_classes corresponde ao número de classes categóricas.

Respondeu 03/12/2015 em 20:17
fonte usuário

Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more