KG/CrimeKgAssitant
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update question_classify.py

Browse Source
This commit is contained in:
liuhuanyong 2018-11-18 22:40:03 +08:00 committed by GitHub
parent 8b336b25a5
commit 9729fa6958
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
question_classify.py
View File

@ -112,8 +112,8 @@ class QuestionClassify(object):
model.add(Conv1D(128, 3, activation='relu')) model.add(Conv1D(128, 3, activation='relu'))
model.add(GlobalAveragePooling1D()) model.add(GlobalAveragePooling1D())
model.add(Dropout(0.5)) model.add(Dropout(0.5))
model.add(Dense(13, activation='sigmoid')) model.add(Dense(13, activation='softmax'))
model.compile(loss='binary_crossentropy', model.compile(loss='categorical_crossentropy',
optimizer='rmsprop', optimizer='rmsprop',
metrics=['accuracy']) metrics=['accuracy'])
model.summary() model.summary()