word_embedding.py

https://github.com/cvxgrp/pymde/blob/main/examples/word_embedding.py