Skip to content

dthiagarajan/word2vec-pytorch

Repository files navigation

word2vec-pytorch

Implementation of word2vec in PyTorch, including both the continuous bag-of-words model and the skipgram model. Subsampling of words is included, as well as negative sampling.