Keras LSTM çok kategoride sınıflandırılması sadece 1 kategori, tahmin - nasıl düzeltilir?

oy
1

Ben etiket eşit sayıda olan bir metin veri kümesi var - 0,1,2,3,4. Koştum Keras binary classificationörneği LSTMbenim veri kümesi ile kendi web sitesinde (imdb örnek) ve derleme hattı olarak değiştirildi

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

Ama model sadece 1 kategori, tahmin 20%sürekli doğruluk.

gerektiği gibi bunu bana / ayarları değiştirme düzeltmek yardım eder misiniz?

Oluştur 29/05/2015 saat 04:53
kaynak kullanıcı
Diğer dillerde...                            


1 cevaplar

oy
6

Sen ile değiştirmeniz gerekir

model.add(Dense(nb_classes))

burada nb_classes kategorik sınıfların sayısına karşılık gelir.

Cevap 03/12/2015 saat 22:17
kaynak kullanıcı

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