sentiment-analysis - 超腾开源标签

本文使用PyTorch实现IMDB电影评论情感分析二分类任务，比较了LSTM、CNN和Transformer三种神经网络模型。介绍了数据加载、构建词表、Dataset类和批处理填充的完整流程。实验结果表明，双向LSTM表现最佳，10个epoch时测试准确率达88.93%；CNN训练速度快且不易过拟合，5个epoch达86.68%准确率；Transformer因数据集规模有限表现中等。使用GloVe预训练词向量可提升LSTM性能至88.76%。文章还提供了模型代码实现、训练流程和优化建议，完整代码可在GPU环境下10-20分钟完成训练。