Fix misleading comment in babi_memnn.py
Output of question_encoder is 3D : sequence of vectors, not single vector.
This commit is contained in:
parent
05aecbc0bc
commit
d322785543
@ -155,7 +155,7 @@ input_encoder_m.add(Embedding(input_dim=vocab_size,
|
|||||||
output_dim=64,
|
output_dim=64,
|
||||||
input_length=story_maxlen))
|
input_length=story_maxlen))
|
||||||
# output: (samples, story_maxlen, embedding_dim)
|
# output: (samples, story_maxlen, embedding_dim)
|
||||||
# embed the question into a single vector
|
# embed the question into a sequence of vectors
|
||||||
question_encoder = Sequential()
|
question_encoder = Sequential()
|
||||||
question_encoder.add(Embedding(input_dim=vocab_size,
|
question_encoder.add(Embedding(input_dim=vocab_size,
|
||||||
output_dim=64,
|
output_dim=64,
|
||||||
|
Loading…
Reference in New Issue
Block a user