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