1. 首页
  2. 编程语言
  3. C
  4. pytorch Gensim word2vec IMDB

pytorch Gensim word2vec IMDB

上传者: 2023-01-03 15:02:48上传 PY文件 4.61KB 热度 11次
Gensim中包含了Word2Vec模型的API.Word2Vec期望输入是进过分词的句子列表即是某个二维数组.该模型API有多个参数可以调整包括词向量的维度vector size扫描句子的窗口大小window训练采用的算法sg和迭代遍历语料库的次数epochs等.训练使用的数据是经过处理的CSV格式的影评数据集IMDB.
下载地址
用户评论