text classification

Python 2025-07-26

pytorch中的text classification

这些代码是用于text classification的不同中性模型的Pytorch实现,包括CNN,LSTM(RNN),C-LSTM。每个型号都有一个自我发挥版本。所使用的数据集是亚马逊评论数据集。

要求

  • Python(> = 3.6)
  • pytorch(> = 1.0)
  • spacy(> = 2.0)
  • torchText(> = 0.4)

参考

  • Ben Trevett的Pytorch情感分析(NLP初学者的出色教程)
  • 句子分类的卷积神经网络
  • 用于文本分类的C-LSTM神经网络
  • [自我注意]
下载源码

通过命令行克隆项目:

git clone https://github.com/luciusssss/text-classification.git