Update question_classify.py

2018-11-18 22:40:03 +08:00 · 2018-11-18 22:40:03 +08:00 · 9729fa6958
commit 9729fa6958
parent 8b336b25a5
1 changed files with 2 additions and 2 deletions
--- a/question_classify.py
+++ b/question_classify.py
@ -112,8 +112,8 @@ class QuestionClassify(object):
        model.add(Conv1D(128, 3, activation='relu'))
        model.add(GlobalAveragePooling1D())
        model.add(Dropout(0.5))
-        model.add(Dense(13, activation='sigmoid'))
-        model.compile(loss='binary_crossentropy',
+        model.add(Dense(13, activation='softmax'))
+        model.compile(loss='categorical_crossentropy',
                      optimizer='rmsprop',
                      metrics=['accuracy'])
        model.summary()