text-classification-nlp-pytorch

Text-classification-nlp-pytorch is used for exploring text classification methods with pytorch.

文本分类、半监督、小样本学习、样本增强

基础文本分类-Base

基础文本分类

基于Bert文本分类

Base Bert Classification

简单bert文本分类

cd train && python train_norm.py

Add Sentence/Encoder Embedding Mixup

添加Embedding Mixup后的文本分类

cd train && python train_mixup.py

半监督学习

PI-Model

cd train && python train_pi_model.py

Temporal-Ensemble

cd train && python train_temporal_ensembling.py

参考文献

- [1] [Augmenting Data with Mixup for Sentence Classification: An Empirical Study](https://arxiv.org/pdf/1905.08941.pdf)

- [2] [Temporal Ensembling for Semi-Supervised Learning](https://arxiv.org/abs/1610.02242)