tex2vec

Skip-gram word2vec with negative sampling, implemented with NumPy.

Quickstart

pip install numpy pyyaml

# train and save
python main.py --train

# load saved model and run demo queries
python main.py --evaluate

# train then immediately evaluate
python main.py --train --evaluate

# custom model path
python main.py --train --model models/my_run.npz