Adding some comments to word2vec.py