Skip to content

Code for the paper Joint Learning of Hyperbolic Label Embeddings for Hierarchical Multi-label Classification (EACL '21)

License

Notifications You must be signed in to change notification settings

soumyac1999/hyperbolic-label-emb-for-hmc

Repository files navigation

Joint Learning of Hyperbolic Label Embeddings for Hierarchical Multi-label Classification

Joint Learning of Hyperbolic Label Embeddings for Hierarchical Multi-label Classification
Soumya Chatterjee*, Ayush Maheshwari*, Ganesh Ramakrishnan and Saketha Nath Jagaralpudi
European Chapter of the Association for Computational Linguistics (EACL) 2021

Requirements

environment.yml has the depedencies

Download glove.6B.300d.txt from here in GloVe folder

Please refer to HiLAP for the dataset instructions. Put the required dataset files in folders rcv1, yelp or nyt and run data_utils/gen_json_<dataset>.py for preprocessing the data.

Run

Run main.py using the arguments --exp_name
--flat for Model_flt
--cascaded_step1 and --cascaded_step2 for Model_cas
--joint for Model_jnt

Specify the dataset using --dataset

For examples, please refer Synthetic/all_expts.sh.

Acknowledgement

Citation:

@inproceedings{chatterjee-etal-2021-joint,
    title = "Joint Learning of Hyperbolic Label Embeddings for Hierarchical Multi-label Classification",
    author = "Chatterjee, Soumya and Maheshwari, Ayush and Ramakrishnan, Ganesh and Jagaralpudi, Saketha Nath",
    booktitle = "Proceedings of the 16th Conference of the European Chapter of the Association for Computational Linguistics: Main Volume",
    year = "2021",
    publisher = "Association for Computational Linguistics",
    url = "https://www.aclweb.org/anthology/2021.eacl-main.247",
    pages = "2829--2841",
}

About

Code for the paper Joint Learning of Hyperbolic Label Embeddings for Hierarchical Multi-label Classification (EACL '21)

Topics

Resources

License

Stars

Watchers

Forks