diff options
| author |  Vijay Vasudevan <vrv@google.com> | 2017-02-17 17:05:49 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-02-17 17:23:48 -0800 |
| commit | 93a975e114ee1c35f01ed3bdd47170e6f7129014 (patch) | |
| tree | e34255aff698fe6a4a586e7940337fd278947f58 /tensorflow/examples/tutorials/word2vec/word2vec_basic.py | |
| parent | eb9624017a0040e805fda622a5f9ec6681e24246 (diff) | |
Merge changes from github.
Change: 147897309
Diffstat (limited to 'tensorflow/examples/tutorials/word2vec/word2vec_basic.py')
| -rw-r--r-- | tensorflow/examples/tutorials/word2vec/word2vec_basic.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/examples/tutorials/word2vec/word2vec_basic.py b/tensorflow/examples/tutorials/word2vec/word2vec_basic.py index bc502edd8b..8dcd3bf37a 100644 --- a/tensorflow/examples/tutorials/word2vec/word2vec_basic.py +++ b/tensorflow/examples/tutorials/word2vec/word2vec_basic.py @@ -112,6 +112,8 @@ def generate_batch(batch_size, num_skips, skip_window): labels[i * num_skips + j, 0] = buffer[target] buffer.append(data[data_index]) data_index = (data_index + 1) % len(data) + # Backtrack a little bit to avoid skipping words in the end of a batch + data_index = (data_index + len(data) - span) % len(data) return batch, labels batch, labels = generate_batch(batch_size=8, num_skips=2, skip_window=1) |