/Text-Classification-Pytorch

text classification model with pytorch

Primary LanguagePythonMIT LicenseMIT

Text-Classification-Pytorch

Neural text classification models implemented by pytorch.

Model list

  • textCNN
  • RCNN(Recurrent CNN)
  • Attention RNN
  • Memory Network

NOTE

Reference

  • Kim, Convolutional Neural Networks for Sentence Classification
  • Lat et al. , Recurrent Convolutional Neural Networks for Text Classification

Licence

This code is under MIT Licence