fix sampling table lookup
This commit is contained in:
parent
1aaab7a228
commit
755728dbb0
@ -69,7 +69,7 @@ def skipgrams(sequence, vocabulary_size,
|
|||||||
if not wi:
|
if not wi:
|
||||||
continue
|
continue
|
||||||
if sampling_table is not None:
|
if sampling_table is not None:
|
||||||
if sampling_table[i] < random.random():
|
if sampling_table[wi] < random.random():
|
||||||
continue
|
continue
|
||||||
|
|
||||||
window_start = max(0, i-window_size)
|
window_start = max(0, i-window_size)
|
||||||
|
Loading…
Reference in New Issue
Block a user