diff --git a/entities/learn_e2v/minibatch_a.lua b/entities/learn_e2v/minibatch_a.lua index b8bfd71..36ec1e5 100644 --- a/entities/learn_e2v/minibatch_a.lua +++ b/entities/learn_e2v/minibatch_a.lua @@ -102,6 +102,7 @@ function process_one_line(line, minibatch, mb_index) -- Still empty ? Get some random words then. if num_positive_words_this_iter == 0 then table.insert(positive_words_in_this_iter, get_word_from_id(random_unigram_at_unig_power_w_id())) + num_positive_words_this_iter = num_positive_words_this_iter + 1 end end @@ -153,4 +154,4 @@ function minibatch_to_correct_type(minibatch) minibatch[1][2] = correct_type(minibatch[1][2]) minibatch[1][3] = correct_type(minibatch[1][3]) minibatch[3][1] = correct_type(minibatch[3][1]) -end \ No newline at end of file +end