├── .gitignore ├── README.md ├── data ├── .DS_Store ├── DB15K │ ├── entities.txt │ ├── relations.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── MKG-W │ ├── .DS_Store │ ├── entities.txt │ ├── entity2id.txt │ ├── relations.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt └── MKG-Y │ ├── entities.txt │ ├── entity2id.txt │ ├── relations.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── dataset.py ├── merge_tokens.py ├── model_mygo.py ├── model_new.py ├── requirements.txt ├── resource └── model.png ├── run.sh ├── save_token_embeddings.py ├── tokens ├── .DS_Store ├── DB15K-textual-llama.json ├── DB15K-textual-roberta.json ├── DB15K-textual-v2.json ├── DB15K-textual.json ├── DB15K-visual-vqgan.json.zip ├── DB15K-visual.json.zip ├── MKG-W-textual.json ├── MKG-W-visual.json.zip ├── MKG-Y-textual-llama.json ├── MKG-Y-textual.json ├── MKG-Y-visual.json.zip ├── visual.pth └── visual_vqgan.pth ├── train_mygo_fgc.py ├── train_other_tokenizer.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | __pycache__/ -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/README.md -------------------------------------------------------------------------------- /data/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/.DS_Store -------------------------------------------------------------------------------- /data/DB15K/entities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/DB15K/entities.txt -------------------------------------------------------------------------------- /data/DB15K/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/DB15K/relations.txt -------------------------------------------------------------------------------- /data/DB15K/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/DB15K/test.txt -------------------------------------------------------------------------------- /data/DB15K/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/DB15K/train.txt -------------------------------------------------------------------------------- /data/DB15K/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/DB15K/valid.txt -------------------------------------------------------------------------------- /data/MKG-W/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-W/.DS_Store -------------------------------------------------------------------------------- /data/MKG-W/entities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-W/entities.txt -------------------------------------------------------------------------------- /data/MKG-W/entity2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-W/entity2id.txt -------------------------------------------------------------------------------- /data/MKG-W/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-W/relations.txt -------------------------------------------------------------------------------- /data/MKG-W/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-W/test.txt -------------------------------------------------------------------------------- /data/MKG-W/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-W/train.txt -------------------------------------------------------------------------------- /data/MKG-W/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-W/valid.txt -------------------------------------------------------------------------------- /data/MKG-Y/entities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-Y/entities.txt -------------------------------------------------------------------------------- /data/MKG-Y/entity2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-Y/entity2id.txt -------------------------------------------------------------------------------- /data/MKG-Y/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-Y/relations.txt -------------------------------------------------------------------------------- /data/MKG-Y/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-Y/test.txt -------------------------------------------------------------------------------- /data/MKG-Y/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-Y/train.txt -------------------------------------------------------------------------------- /data/MKG-Y/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/data/MKG-Y/valid.txt -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/dataset.py -------------------------------------------------------------------------------- /merge_tokens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/merge_tokens.py -------------------------------------------------------------------------------- /model_mygo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/model_mygo.py -------------------------------------------------------------------------------- /model_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/model_new.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/requirements.txt -------------------------------------------------------------------------------- /resource/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/resource/model.png -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/run.sh -------------------------------------------------------------------------------- /save_token_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/save_token_embeddings.py -------------------------------------------------------------------------------- /tokens/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/.DS_Store -------------------------------------------------------------------------------- /tokens/DB15K-textual-llama.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/DB15K-textual-llama.json -------------------------------------------------------------------------------- /tokens/DB15K-textual-roberta.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/DB15K-textual-roberta.json -------------------------------------------------------------------------------- /tokens/DB15K-textual-v2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/DB15K-textual-v2.json -------------------------------------------------------------------------------- /tokens/DB15K-textual.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/DB15K-textual.json -------------------------------------------------------------------------------- /tokens/DB15K-visual-vqgan.json.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/DB15K-visual-vqgan.json.zip -------------------------------------------------------------------------------- /tokens/DB15K-visual.json.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/DB15K-visual.json.zip -------------------------------------------------------------------------------- /tokens/MKG-W-textual.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/MKG-W-textual.json -------------------------------------------------------------------------------- /tokens/MKG-W-visual.json.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/MKG-W-visual.json.zip -------------------------------------------------------------------------------- /tokens/MKG-Y-textual-llama.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/MKG-Y-textual-llama.json -------------------------------------------------------------------------------- /tokens/MKG-Y-textual.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/MKG-Y-textual.json -------------------------------------------------------------------------------- /tokens/MKG-Y-visual.json.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/MKG-Y-visual.json.zip -------------------------------------------------------------------------------- /tokens/visual.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/visual.pth -------------------------------------------------------------------------------- /tokens/visual_vqgan.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/tokens/visual_vqgan.pth -------------------------------------------------------------------------------- /train_mygo_fgc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/train_mygo_fgc.py -------------------------------------------------------------------------------- /train_other_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/train_other_tokenizer.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjukg/MyGO/HEAD/utils.py --------------------------------------------------------------------------------