pytorch Gensim word2vec IMDB

Gensim中包含了Word2Vec模型的API.Word2Vec期望输入是进过分词的句子列表即是某个二维数组.该模型API有多个参数可以调整包括词向量的维度vector size扫描句子的窗口大小window训练采用的算法sg和迭代遍历语料库的次数epochs等.训练使用的数据是经过处理的CSV格式的影评数据集IMDB.