Label-aware Document Representation via Hybrid Attention for Extreme Multi-Label Text Classification

pytorch implement for paper Label-aware Document Representation via Hybrid Attention for Extreme Multi-Label Text Classification.

Set up the dataset

download the dataset and Glove embedding from here.

Train the model

run python hybrid_xml.py.