0ff700abccc71ceb0794ddc8e77945e178f10599,examples/babi_memnn.py,,,#,101

Before Change

print("Vectorizing the word sequences...")

word_idx = dict((c, i + 1) for i, c in enumerate(vocab))
inputs_train, queries_train, answers_train = vectorize_stories(train_stories,
                                                               word_idx,
                                                               story_maxlen,
                                                               query_maxlen)
inputs_test, queries_test, answers_test = vectorize_stories(test_stories,
                                                            word_idx,
                                                            story_maxlen,
                                                            query_maxlen)

print("-")
print("inputs: integer tensor of shape (samples, max_length)")
print("inputs_train shape:", inputs_train.shape)

After Change

word_idx = dict((c, i + 1) for i, c in enumerate(vocab))
inputs_train, queries_train, answers_train = vectorize_stories(train_stories)
inputs_test, queries_test, answers_test = vectorize_stories(test_stories)

print("-")
print("inputs: integer tensor of shape (samples, max_length)")
print("inputs_train shape:", inputs_train.shape)

Instances