diff options
author | Illia Polosukhin <ilblackdragon@gmail.com> | 2016-04-18 17:56:51 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-04-18 19:03:29 -0700 |
commit | 5c9bc51857bc0c330d3ab976871ee3509647d1e7 (patch) | |
tree | a58def7cbf316c6e091b3b36657f120f1388ec54 /tensorflow/examples/tutorials/word2vec/word2vec_basic.py | |
parent | fc432e37a7ddd408ff09a7b90b1c4cd5af1b134e (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.py | 6 |
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. |