word2vec-pytorch icon indicating copy to clipboard operation
word2vec-pytorch copied to clipboard

Implementation of word2vec in PyTorch

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.