Counting 3,663 Big Data & Machine Learning Frameworks, Toolsets, and Examples...
Suggestion? Feedback? Tweet @stkim1

Word2vec-PyTorch

Warning: this repo is poseted two years ago and it's not compatible with the current pytorch (v1.0.0). I will update it as soon as possible

This is the implementation of word2vec based on PyTorch.

Run it with

python word2vec.py zhihu.txt word_embedding.txt

I write a blog about the word2vec based on PyTorch.