From 437aeb55cc89fade6e386205b30148bc21471bb1 Mon Sep 17 00:00:00 2001 From: Castiel Date: Fri, 23 Nov 2018 07:31:57 +1030 Subject: [PATCH 1/2] Minor change in word2vec_basic tutorial --- tensorflow/examples/tutorials/word2vec/word2vec_basic.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/examples/tutorials/word2vec/word2vec_basic.py b/tensorflow/examples/tutorials/word2vec/word2vec_basic.py index b09ee997689..bbcfc32098a 100644 --- a/tensorflow/examples/tutorials/word2vec/word2vec_basic.py +++ b/tensorflow/examples/tutorials/word2vec/word2vec_basic.py @@ -91,7 +91,7 @@ vocabulary_size = 50000 def build_dataset(words, n_words): """Process raw inputs into a dataset.""" - count = [['UNK', -1]] + count = [('UNK', -1)] count.extend(collections.Counter(words).most_common(n_words - 1)) dictionary = dict() for word, _ in count: @@ -125,6 +125,7 @@ data_index = 0 # Step 3: Function to generate a training batch for the skip-gram model. def generate_batch(batch_size, num_skips, skip_window): + global data global data_index assert batch_size % num_skips == 0 assert num_skips <= 2 * skip_window From 795f16f1fbf7b2018475c81c43e1050a1f87ce8e Mon Sep 17 00:00:00 2001 From: Mark Daoust Date: Tue, 11 Dec 2018 13:52:38 -0800 Subject: [PATCH 2/2] remove `global data` --- tensorflow/examples/tutorials/word2vec/word2vec_basic.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tensorflow/examples/tutorials/word2vec/word2vec_basic.py b/tensorflow/examples/tutorials/word2vec/word2vec_basic.py index bbcfc32098a..77889effc8e 100644 --- a/tensorflow/examples/tutorials/word2vec/word2vec_basic.py +++ b/tensorflow/examples/tutorials/word2vec/word2vec_basic.py @@ -125,7 +125,6 @@ data_index = 0 # Step 3: Function to generate a training batch for the skip-gram model. def generate_batch(batch_size, num_skips, skip_window): - global data global data_index assert batch_size % num_skips == 0 assert num_skips <= 2 * skip_window