n0obcoder / Skip-Gram-Model-PyTorch Public

Notifications You must be signed in to change notification settings
Fork 13
Star 53

PyTorch implementation of the Word2Vec (Skip-Gram Model) and visualizing the trained embeddings using TSNE

53 stars 13 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 8 Commits
README.md		README.md
config.py		config.py
datasets.py		datasets.py
main.py		main.py
model.py		model.py
test.py		test.py
utils_modified.py		utils_modified.py

Repository files navigation

Skip-Gram-Model-PyTorch

PyTorch implementation of the word2vec (skip-gram model) and visualization of the trained embeddings using TSNE !

Requirements

torch >= 1.4
numpy >= 1.18
matplotlib
tqdm
nltk
gensim

Training

python main.py

Inference/Testing

python test.py

About

PyTorch implementation of the Word2Vec (Skip-Gram Model) and visualizing the trained embeddings using TSNE

python nlp natural-language-processing word2vec pytorch neural-networks skip-gram tsne skipgram wordembeddings

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%