https://github.com/songjiang0909/BoxTaxo

Code for paper "A Single Vector Is Not Enough: Taxonomy Expansion via Box Embeddings"
https://github.com/songjiang0909/BoxTaxo

Last synced: about 1 month ago
JSON representation

Code for paper "A Single Vector Is Not Enough: Taxonomy Expansion via Box Embeddings"

• Host: GitHub
• URL: https://github.com/songjiang0909/BoxTaxo
• Owner: songjiang0909
• Created: 2023-02-10T01:30:26.000Z (almost 2 years ago)
• Default Branch: main
• Last Pushed: 2023-05-28T06:53:04.000Z (over 1 year ago)
• Last Synced: 2024-08-05T03:01:53.701Z (5 months ago)
• Language: Python
• Homepage:
• Size: 123 KB
• Stars: 12
• Watchers: 1
• Forks: 2
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• awesome-taxonomy - https://github.com/songjiang0909/BoxTaxo

README

        
BoxTaxo

-----------------

This is the implementation of our paper "[A Single Vector Is Not Enough: Taxonomy Expansion via  Box Embeddings](https://songjiang0909.github.io/pdf/boxtaxo.pdf)", published in WWW'23.

Data

-----------------

The original data used could be access from SemEval-2016 Task 13: Taxonomy Extraction Evaluation.

We also provide our processed data under the `data` folder.  Credits to [repo](https://github.com/yueyu1030/STEAM)!

Requirement

----------------------

All required packages could be found in `requirement.txt` (generated by `pip freeze`).

How To Run?

----------------------

```bash

cd ./src

python main.py

```

Key Arguments Interpretation:

- `--dataset`: Dataset option: environment or science.

- `--embed_size`: Dimension of box embeddings.

- `--margin` : Margin for containing loss ($\delta$ in paper).

- `--epsilon`: Margin for negative contain ($\epsilon$ in paper).

- `--size`:  Minimum box ($\phi$ in paper).

The results (log, predictions, learned boxes) are stored under the `result` folder.

Contact

----------------------

Song Jiang 

Bibtex

----------------------

```bibtex

@inproceedings{boxtaxo,

  title={A Single Vector Is Not Enough: Taxonomy Expansion via  Box Embeddings},

  author={Song Jiang, Qiyue Yao, Qifan Wang, Yizhou Sun},

  booktitle={Proceedings of The Web Conference},

  year={2023}

}

```