The code for "Does Head Label Help for Long-Tailed Multi-Label Text Classification"
If you make use of this code or the HTTN algorithm in your work, please cite the following paper:
@article{xiao2021does,
title={Does Head Label Help for Long-Tailed Multi-Label Text Classification},
author={Xiao, Lin and Zhang, Xiangliang and Jing, Liping and Huang, Chi and Song, Mingyang},
journal={arXiv preprint arXiv:2101.09704},
year={2021}
}
Requirements: Ubuntu 16.0.4;
Python 3.6;
Pytorch 1.0.0;
mxnet 1.2.1;
scikit-learn 0.20.1;
Reproducibility: We provide the processed dataset AAPD, put them in the folder./data/
Processed data download: https://drive.google.com/file/d/1QoqcJkZBHsDporttTxaYWOM_ExSn7-Dz/view