Training a Word2Vec using TensorFlow