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
Download the dataset and unzip it in project folder.
Modify the data path variable DATA_PATH in LCKGE/src/datasets.py file with the dataset path
Run train_learn_FB15K-237.sh, train_learn_FB15K.sh or train_learn_WN9.sh .sh files