
PyTorch+Gensim+Word2Vec在IMDB数据上的应用
5星
- 浏览量: 0
- 大小:None
- 文件类型:PY
简介:
本项目运用PyTorch框架结合Gensim库中的Word2Vec模型,在IMDb电影评论数据集上进行情感分析实验,探索词嵌入技术在文本分类任务中的效果。
Gensim 包含了 Word2Vec 模型的 API。Word2Vec 需要输入经过分词的句子列表,即是一个二维数组。该模型 API 提供多个可调参数,包括词向量维度(vector_size)、扫描窗口大小(window)、训练算法类型(sg)以及遍历语料库次数(epochs)。用于训练的数据是处理过的 CSV 格式的 IMDb 影评数据集。
全部评论 (0)
还没有任何评论哟~


