Update question_classify_train.py

2018-11-18 22:41:02 +08:00 · 2018-11-18 22:41:02 +08:00 · c1b8befb44
commit c1b8befb44
parent 9729fa6958
1 changed files with 2 additions and 2 deletions
--- a/question_classify_train.py
+++ b/question_classify_train.py
@ -129,8 +129,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()