NLP pytorch LSTM文本情感分类

上传：imagination7357 浏览： 17 推荐： 0 文件：py 大小：11.63KB 上传时间：2023-01-03 15:02:58 版权申诉

基于训练好的语言模型使用gensim的word2vecAPI编写了一个情感分类模型包含一个循环神经网络模型LSTM和一个分类器MLP.首先将一个句子中的每个单词对应的词向量输入循环神经网络得到句子的向量表征.然后将句向量作为分类器的输入输出二元分类预测同样进行loss计算和反向梯度传播训练这里的loss使用交叉熵loss.