nanjusoil/Hybrid_Attention_XML

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

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.