/word2vec_pytorch

Primary LanguagePythonMIT LicenseMIT

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.