aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/examples/tutorials/word2vec/word2vec_basic.py
diff options
context:
space:
mode:
authorGravatar Illia Polosukhin <ilblackdragon@gmail.com>2016-04-18 17:56:51 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-04-18 19:03:29 -0700
commit5c9bc51857bc0c330d3ab976871ee3509647d1e7 (patch)
treea58def7cbf316c6e091b3b36657f120f1388ec54 /tensorflow/examples/tutorials/word2vec/word2vec_basic.py
parentfc432e37a7ddd408ff09a7b90b1c4cd5af1b134e (diff)
Merge changes from github.
Change: 120185825
Diffstat (limited to 'tensorflow/examples/tutorials/word2vec/word2vec_basic.py')
-rw-r--r--tensorflow/examples/tutorials/word2vec/word2vec_basic.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/examples/tutorials/word2vec/word2vec_basic.py b/tensorflow/examples/tutorials/word2vec/word2vec_basic.py
index 6cbe6a0f7e..83bb5dd165 100644
--- a/tensorflow/examples/tutorials/word2vec/word2vec_basic.py
+++ b/tensorflow/examples/tutorials/word2vec/word2vec_basic.py
@@ -81,7 +81,7 @@ def build_dataset(words):
data, count, dictionary, reverse_dictionary = build_dataset(words)
del words # Hint to reduce memory.
print('Most common words (+UNK)', count[:5])
-print('Sample data', data[:10])
+print('Sample data', data[:10], [reverse_dictionary[i] for i in data[:10]])
data_index = 0
@@ -113,8 +113,8 @@ def generate_batch(batch_size, num_skips, skip_window):
batch, labels = generate_batch(batch_size=8, num_skips=2, skip_window=1)
for i in range(8):
- print(batch[i], '->', labels[i, 0])
- print(reverse_dictionary[batch[i]], '->', reverse_dictionary[labels[i, 0]])
+ print(batch[i], reverse_dictionary[batch[i]],
+ '->', labels[i, 0], reverse_dictionary[labels[i, 0]])
# Step 4: Build and train a skip-gram model.