wassname/keras-language-modeling
1
0
Fork 0
mirror of https://github.com/wassname/keras-language-modeling.git synced 2026-09-09 11:25:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

minor changes

Browse Source
This commit is contained in:
codekansas
2016-07-24 21:17:20 -07:00
parent 39dd0caad4
commit b70eb2f7b6
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
seq2seq/answer_to_question.py
+2 -5
View File
@@ -95,10 +95,7 @@ def get_model(question_maxlen, answer_maxlen, vocab_len, n_hidden, load_save=Fal
model.compile(loss='categorical_crossentropy', optimizer='rmsprop', metrics=['accuracy'])
if os.path.exists(model_save) and load_save:
try:
model.load_weights(model_save)
except Exception:
print('A model exists at "%s", but it is not compatible' % model_save)
model.load_weights(model_save)
return model
@@ -191,7 +188,7 @@ if __name__ == '__main__':
print('Generating model...')
model = get_model(question_maxlen=question_maxlen, answer_maxlen=answer_maxlen, vocab_len=len(qa.vocab) + 1,
n_hidden=256, load_save=False)
n_hidden=256, load_save=True)
print('Training model...')
for iteration in range(1, nb_iteration + 1):