NeuralClassifier: An Open-source Neural Hierarchical Multi-label Text Classification Toolkit

Liqun Liu, Funan Mu, Pengyu Li, Xin Mu, Jing Tang, Xingsheng Ai, Ran Fu, Lifeng Wang, Xing Zhou
2019 Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics: System Demonstrations  
In this paper, we introduce NeuralClassifier, a toolkit for neural hierarchical multi-label text classification. NeuralClassifier is designed for quick implementation of neural models for hierarchical multi-label classification task, which is more challenging and common in real-world scenarios. A salient feature is that NeuralClassifier currently provides a variety of text encoders, such as FastText, TextCNN, TextRNN, RCNN, VDCNN, DPCNN, DRNN, AttentiveConvNet and Transformer encoder, etc. It
more » ... so supports other text classification scenarios, including binary-class and multiclass classification. Built on PyTorch 1 , the core operations are calculated in batch, making the toolkit efficient with the acceleration of GPU. Experiments show that models built in our toolkit achieve comparable performance with reported results in the literature. * Equal contribution 1 https://pytorch.org/
doi:10.18653/v1/p19-3015 dblp:conf/acl/LiuMLMTAFWZ19 fatcat:7vcqanwnovcupni7ii25rayov4