From 93a975e114ee1c35f01ed3bdd47170e6f7129014 Mon Sep 17 00:00:00 2001 From: Vijay Vasudevan Date: Fri, 17 Feb 2017 17:05:49 -0800 Subject: Merge changes from github. Change: 147897309 --- tensorflow/examples/tutorials/word2vec/word2vec_basic.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'tensorflow/examples/tutorials/word2vec/word2vec_basic.py') 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) -- cgit v1.2.3