PengFoo / word2vec-pytorch

A Skip-Gram model of Word2vec implemented in pytorch.
9 stars 10 forks source link

Not using subsampling #2

Open AndrewLim1990 opened 4 years ago

AndrewLim1990 commented 4 years ago

Hi! Thanks for this repo - super useful.

I think you may have forgotten to use your subsampling table sub_sampling_table:

https://github.com/PengFoo/word2vec-pytorch/blob/99269ae8c50ad3e0ebc9b7a034213546208a008f/word2vec/data_handler.py#L51

I see it being defined but I can't find any references to it.