- Notifications
You must be signed in to change notification settings - Fork13
n0obcoder/Skip-Gram-Model-PyTorch
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
PyTorch implementation of the word2vec (skip-gram model) and visualization of the trained embeddings using TSNE !
My TensorFlow implemntation of Skip-Gram Model can be foundhere.
- torch >= 1.4
- numpy >= 1.18
- matplotlib
- tqdm
- nltk
- gensim
python main.pytensorboard --logdir <PATH_TO_TENSORBOARD_EVENTS_FILE>NOTE: By default,PATH_TO_TENSORBOARD_EVENTS_FILE is set toSUMMARY_DIR in config.py
python test.py| war | india | crime | guitar | movies | desert | physics | religion | football | computer |
|---|---|---|---|---|---|---|---|---|---|
| fight | europe | despite | band | movie | region | theory | religious | team | program |
| battle | central | help | play | series | along | mathematics | christian | win | systems |
| army | western | seek | record | show | western | mathematical | regard | sport | available |
| force | indian | challenge | piece | film | southern | study | tradition | club | design |
| ally | part | fail | star | feature | plain | science | christianity | league | information |
Check out my blog post onword2vechere.
About
PyTorch implementation of the Word2Vec (Skip-Gram Model) and visualizing the trained embeddings using TSNE
Topics
Resources
Uh oh!
There was an error while loading.Please reload this page.
Stars
Watchers
Forks
Releases
No releases published
Packages0
No packages published