code of paper "Multimodal Knowledge Graph Embeddings via Lorentz-based Contrastive Learning"

cite: R. Liu, Z. Cao, Z. Wu, Q. Xu and Q. Huang, "Multimodal Knowledge Graph Embeddings via Lorentz-based Contrastive Learning," 2024 IEEE International Conference on Multimedia and Expo (ICME), Niagara Falls, ON, Canada, 2024, pp. 1-6, doi: 10.1109/ICME57554.2024.10687608.

Multimodal Knwoledge Graph dataset https://huggingface.co/datasets/Ruizhou22/mmkg/blob/main/mmkg_data.zip

1. Download the dataset and unzip it in project folder.

2. Modify the data path variable DATA_PATH in LCKGE/src/datasets.py file with the dataset path

3. Run train_learn_FB15K-237.sh, train_learn_FB15K.sh or train_learn_WN9.sh .sh files