TextClassification icon indicating copy to clipboard operation
TextClassification copied to clipboard

All kinds of neural text classifiers implemented by Keras

Text Classification

All kinds of neural text classififers implemented by Keras (tensorflow backend).

Models

  • TextCNN, EMNLP2014
    Kim et al. Convolutional Neural Networks for Sentence Classification.

  • DCNN, ACL2014
    Kalchbrenner et al. A Convolutional Neural Network for Modelling Sentences

  • RCNN, AAAI2015
    Lai et al. Recurrent Convolutional Neural Networks for Text Classification.

  • HAN, NAACL-HLT2016
    Yang et al. Hierarchical Attention Networks for Document Classification.

  • DPCNN, ACL2017
    Johnson et al. Deep Pyramid Convolutional Neural Networks for Text Categorization.

  • VDCNN, EACL2017
    Conneau et al. Very Deep Convolutional Networks for Text Classification.

  • MultiTextCNN
    Extension of textcnn, stacking multiple cnns with the same filter size.

  • BiLSTM
    Bidirectional lstm + max pooling over time.

  • RNNCNN
    Bidirectional gru + conv + max pooling & avg pooling.

  • CNNRNN
    conv + max pooling + Bidirectional gru + max pooling over time.