modify architecture of word2vec