Skip to content

Extremely simple and fast word2vec implementation with Negative Sampling + Sub-sampling

Notifications You must be signed in to change notification settings

RomanShen/word2vec-pytorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Word2vec Pytorch

Fast word2vec implementation at competitive speed compared with fasttext. The slowest part is the python data loader. Indeed, Python isn't the fastest programming language, maybe you can improve the code :)

Advantages

  • Easy to understand, solid code
  • Easy to extend for new experiments
  • You can try advanced learning optimizers, with new learning technics
  • GPU support

Supported features

  • Skip-gram
  • Batch update
  • Cosine Annealing
  • Negative Sampling
  • Sub-sampling of frequent word

About

Extremely simple and fast word2vec implementation with Negative Sampling + Sub-sampling

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%