PyTorch-Text-Classifers

Collect and Implement Neural Text Classification Models by Pytorch

Requirement

• GPU
• CUDA
• PyTorch

Implemented Models

• FastText [Joulin et al. 2017]
• MLP
• Stacked Convolutional Neural Network
• Parallel Convolutional Neural Network [Kim 2014]
• LSTM (Bidirectional/Unidirectional/Self-attention Supported)
• Transformer [Vaswani 2017] (Only Encoder + Softmax for classification)

Guidance

For details and quick start, see the README in the corresponding folder to each model.

TODO LIST

• Gated Convolutional Network [Gehring 2017]
• Transformer with Pretraining (BERT) [Devlin 2018]