Word2Vec implementation in pure Numpy

Two variants of skip-gram

softmax skip-gram and sigmoid skip-gram with negative sampling

softmax skip-gram training loop can be found in skip_gram.py negative sampling training loop can be found in negative_sampling.py

---

how to run this project?

uv sync
uv run main.py

or

pip install -r requirements.txt
pytnon main.py

Regular Skip-gram with Softmax

uv run skip_gram.py

Skip-gram with negative sampling

uv run negative_sampling.py

---