Word Embeddings Python