fix sampling table lookup

2015-05-25 11:10:18 +02:00 · 2015-05-25 11:10:18 +02:00 · 755728dbb0
commit 755728dbb0
parent 1aaab7a228
1 changed files with 1 additions and 1 deletions
--- a/keras/preprocessing/sequence.py
+++ b/keras/preprocessing/sequence.py
@ -69,7 +69,7 @@ def skipgrams(sequence, vocabulary_size,
        if not wi:
            continue
        if sampling_table is not None:
-            if sampling_table[i] < random.random():
+            if sampling_table[wi] < random.random():
                continue

        window_start = max(0, i-window_size)