├── .gitignore ├── LICENSE ├── MKG ├── data │ ├── __init__.py │ ├── base_data_module.py │ ├── data_module.py │ └── processor.py ├── dataset │ ├── FB15k-237 │ │ ├── 8shot-42-train.tsv │ │ ├── dev.tsv │ │ ├── entities.txt │ │ ├── entity2text.txt │ │ ├── entity2textlong.txt │ │ ├── relation2text.txt │ │ ├── relations.txt │ │ ├── test.tsv │ │ └── train.tsv │ ├── FB15k-images │ │ ├── m.0_00 │ │ │ └── images │ │ ├── m.0_2v │ │ │ └── images │ │ ├── m.0_3cs │ │ │ └── images │ │ └── m.0__wm │ │ │ └── images │ ├── WN18 │ │ ├── dev.tsv │ │ ├── entities.txt │ │ ├── entity2id.txt │ │ ├── entity2text.txt │ │ ├── relation2id.txt │ │ ├── relation2text.txt │ │ ├── relations.txt │ │ ├── test.tsv │ │ ├── test2id.txt │ │ ├── train.tsv │ │ ├── train2id.txt │ │ ├── valid2id.txt │ │ └── wordnet-mlj12-definitions.txt │ └── wn18-images │ │ ├── n00004475 │ │ └── images │ │ ├── n00006484 │ │ └── images │ │ ├── n00007846 │ │ └── images │ │ └── n00015388 │ │ └── images ├── lit_models │ ├── __init__.py │ ├── base.py │ ├── transformer.py │ └── utils.py ├── main.py ├── models │ ├── __init__.py │ ├── model.py │ ├── modeling_clip.py │ ├── modeling_unimo.py │ └── utils.py └── scripts │ ├── fb15k-237-image.sh │ ├── pretrain_fb15k-237-image.sh │ ├── pretrain_wn18-images.sh │ ├── wn18-images.sh │ └── wn19-image-train.cfg ├── MNER ├── models │ ├── modeling_clip.py │ ├── modeling_unimo.py │ └── unimo_model.py ├── modules │ ├── metrics.py │ └── train.py ├── processor │ └── datasets.py ├── run.py ├── run_mner.sh └── utils │ ├── __init__.py │ ├── ner_evaluate.py │ └── utils.py ├── MRE ├── models │ ├── __init__.py │ ├── modeling_clip.py │ ├── modeling_unimo.py │ └── unimo_model.py ├── modules │ ├── metrics.py │ └── train.py ├── processor │ └── dataset.py ├── run.py └── run_mre.sh ├── README.md ├── requirements.txt └── resource └── model.png /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/LICENSE -------------------------------------------------------------------------------- /MKG/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/data/__init__.py -------------------------------------------------------------------------------- /MKG/data/base_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/data/base_data_module.py -------------------------------------------------------------------------------- /MKG/data/data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/data/data_module.py -------------------------------------------------------------------------------- /MKG/data/processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/data/processor.py -------------------------------------------------------------------------------- /MKG/dataset/FB15k-237/8shot-42-train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/FB15k-237/8shot-42-train.tsv -------------------------------------------------------------------------------- /MKG/dataset/FB15k-237/dev.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/FB15k-237/dev.tsv -------------------------------------------------------------------------------- /MKG/dataset/FB15k-237/entities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/FB15k-237/entities.txt -------------------------------------------------------------------------------- /MKG/dataset/FB15k-237/entity2text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/FB15k-237/entity2text.txt -------------------------------------------------------------------------------- /MKG/dataset/FB15k-237/entity2textlong.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/FB15k-237/entity2textlong.txt -------------------------------------------------------------------------------- /MKG/dataset/FB15k-237/relation2text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/FB15k-237/relation2text.txt -------------------------------------------------------------------------------- /MKG/dataset/FB15k-237/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/FB15k-237/relations.txt -------------------------------------------------------------------------------- /MKG/dataset/FB15k-237/test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/FB15k-237/test.tsv -------------------------------------------------------------------------------- /MKG/dataset/FB15k-237/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/FB15k-237/train.tsv -------------------------------------------------------------------------------- /MKG/dataset/FB15k-images/m.0_00/images: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MKG/dataset/FB15k-images/m.0_2v/images: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MKG/dataset/FB15k-images/m.0_3cs/images: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MKG/dataset/FB15k-images/m.0__wm/images: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MKG/dataset/WN18/dev.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/dev.tsv -------------------------------------------------------------------------------- /MKG/dataset/WN18/entities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/entities.txt -------------------------------------------------------------------------------- /MKG/dataset/WN18/entity2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/entity2id.txt -------------------------------------------------------------------------------- /MKG/dataset/WN18/entity2text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/entity2text.txt -------------------------------------------------------------------------------- /MKG/dataset/WN18/relation2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/relation2id.txt -------------------------------------------------------------------------------- /MKG/dataset/WN18/relation2text.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/relation2text.txt -------------------------------------------------------------------------------- /MKG/dataset/WN18/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/relations.txt -------------------------------------------------------------------------------- /MKG/dataset/WN18/test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/test.tsv -------------------------------------------------------------------------------- /MKG/dataset/WN18/test2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/test2id.txt -------------------------------------------------------------------------------- /MKG/dataset/WN18/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/train.tsv -------------------------------------------------------------------------------- /MKG/dataset/WN18/train2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/train2id.txt -------------------------------------------------------------------------------- /MKG/dataset/WN18/valid2id.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/valid2id.txt -------------------------------------------------------------------------------- /MKG/dataset/WN18/wordnet-mlj12-definitions.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/dataset/WN18/wordnet-mlj12-definitions.txt -------------------------------------------------------------------------------- /MKG/dataset/wn18-images/n00004475/images: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MKG/dataset/wn18-images/n00006484/images: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MKG/dataset/wn18-images/n00007846/images: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MKG/dataset/wn18-images/n00015388/images: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /MKG/lit_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/lit_models/__init__.py -------------------------------------------------------------------------------- /MKG/lit_models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/lit_models/base.py -------------------------------------------------------------------------------- /MKG/lit_models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/lit_models/transformer.py -------------------------------------------------------------------------------- /MKG/lit_models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/lit_models/utils.py -------------------------------------------------------------------------------- /MKG/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/main.py -------------------------------------------------------------------------------- /MKG/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import * 2 | 3 | -------------------------------------------------------------------------------- /MKG/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/models/model.py -------------------------------------------------------------------------------- /MKG/models/modeling_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/models/modeling_clip.py -------------------------------------------------------------------------------- /MKG/models/modeling_unimo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/models/modeling_unimo.py -------------------------------------------------------------------------------- /MKG/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/models/utils.py -------------------------------------------------------------------------------- /MKG/scripts/fb15k-237-image.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/scripts/fb15k-237-image.sh -------------------------------------------------------------------------------- /MKG/scripts/pretrain_fb15k-237-image.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/scripts/pretrain_fb15k-237-image.sh -------------------------------------------------------------------------------- /MKG/scripts/pretrain_wn18-images.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/scripts/pretrain_wn18-images.sh -------------------------------------------------------------------------------- /MKG/scripts/wn18-images.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/scripts/wn18-images.sh -------------------------------------------------------------------------------- /MKG/scripts/wn19-image-train.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MKG/scripts/wn19-image-train.cfg -------------------------------------------------------------------------------- /MNER/models/modeling_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MNER/models/modeling_clip.py -------------------------------------------------------------------------------- /MNER/models/modeling_unimo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MNER/models/modeling_unimo.py -------------------------------------------------------------------------------- /MNER/models/unimo_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MNER/models/unimo_model.py -------------------------------------------------------------------------------- /MNER/modules/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MNER/modules/metrics.py -------------------------------------------------------------------------------- /MNER/modules/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MNER/modules/train.py -------------------------------------------------------------------------------- /MNER/processor/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MNER/processor/datasets.py -------------------------------------------------------------------------------- /MNER/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MNER/run.py -------------------------------------------------------------------------------- /MNER/run_mner.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MNER/run_mner.sh -------------------------------------------------------------------------------- /MNER/utils/__init__.py: -------------------------------------------------------------------------------- 1 | import utils -------------------------------------------------------------------------------- /MNER/utils/ner_evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MNER/utils/ner_evaluate.py -------------------------------------------------------------------------------- /MNER/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MNER/utils/utils.py -------------------------------------------------------------------------------- /MRE/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MRE/models/__init__.py -------------------------------------------------------------------------------- /MRE/models/modeling_clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MRE/models/modeling_clip.py -------------------------------------------------------------------------------- /MRE/models/modeling_unimo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MRE/models/modeling_unimo.py -------------------------------------------------------------------------------- /MRE/models/unimo_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MRE/models/unimo_model.py -------------------------------------------------------------------------------- /MRE/modules/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MRE/modules/metrics.py -------------------------------------------------------------------------------- /MRE/modules/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MRE/modules/train.py -------------------------------------------------------------------------------- /MRE/processor/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MRE/processor/dataset.py -------------------------------------------------------------------------------- /MRE/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MRE/run.py -------------------------------------------------------------------------------- /MRE/run_mre.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/MRE/run_mre.sh -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/requirements.txt -------------------------------------------------------------------------------- /resource/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/MKGformer/HEAD/resource/model.png --------------------------------------------------------------------------------