| Quick Start | Website | Paper | Video
- Python 3.10.13
- PyTorch 2.0.1
for detailed information, please see requirements.txt
git clone https://github.com/JohnJiang12138/CMRP.git
cd CMRP
conda create -n CMRP python=3.10.13
conda activate CMRP
pip install -r requirements.txt
During preprocessing, we use T5-Small for embedding.
git lfs install
git clone https://huggingface.co/google-t5/t5-small
Please replace the api_key
and the requests
detail with your customization.
In fileLLM_tasks/t1step2_llm_evaluate.py
, line 9 and line 73.
Here we preprocess the datasets, prepare embeddings and calculate similarities.
cd ./preprocess
sh preprocess_WN18.sh
cd ../REINFORCE
sh run_kg_WN18.sh
Thanks! / 谢谢! / ありがとう! / merci! / 감사! / Danke! / спасибо! / gracias! ...
@inproceedings{jiang2024killing,
title={Killing Two Birds with One Stone: Cross-modal Reinforced Prompting for Graph and Language Tasks},
author={Jiang, Wenyuan and Wu, Wenwei and Zhang, Le and Yuan, Zixuan and Xiang, Jian and Zhou, Jingbo and Xiong, Hui},
booktitle={Proceedings of the 30th ACM SIGKDD Conference on Knowledge Discovery and Data Mining},
pages={1301--1312},
year={2024}
}