├── .gitignore ├── DATASET ├── CN15k │ ├── check.py │ ├── entities.txt │ ├── relations.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── NL27K │ ├── check.py │ ├── entities.txt │ ├── relations.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── UFB15K237 │ ├── entity_id.tsv │ ├── relation_id.tsv │ ├── test2id.tsv │ ├── train2id.tsv │ └── valid2id.tsv ├── UWN18RR │ ├── entity_id.tsv │ ├── relation_id.tsv │ ├── test2id.tsv │ ├── train2id.tsv │ └── valid2id.tsv ├── confidece_analyse.ipynb └── histogram.eps ├── LICENSE ├── README.md ├── bert-base-uncased ├── config.json ├── pytorch_model.bin ├── tokenizer.json ├── tokenizer_config.json └── vocab.txt ├── confidence_prediction.py ├── decode_rules └── dataset │ ├── CN15K │ ├── cn15k_path_test.txt │ ├── cn15k_path_train.txt │ └── cn15k_path_valid.txt │ └── NL27K │ ├── nl27k_path_test.txt │ ├── nl27k_path_train.txt │ └── nl27k_path_valid.txt ├── transformer ├── Layers.py ├── Models.py ├── Modules.py ├── Optim.py ├── SubLayers.py ├── Translator.py └── dataset.py ├── translate_decode.py └── translate_train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/.gitignore -------------------------------------------------------------------------------- /DATASET/CN15k/check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/CN15k/check.py -------------------------------------------------------------------------------- /DATASET/CN15k/entities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/CN15k/entities.txt -------------------------------------------------------------------------------- /DATASET/CN15k/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/CN15k/relations.txt -------------------------------------------------------------------------------- /DATASET/CN15k/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/CN15k/test.txt -------------------------------------------------------------------------------- /DATASET/CN15k/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/CN15k/train.txt -------------------------------------------------------------------------------- /DATASET/CN15k/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/CN15k/valid.txt -------------------------------------------------------------------------------- /DATASET/NL27K/check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/NL27K/check.py -------------------------------------------------------------------------------- /DATASET/NL27K/entities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/NL27K/entities.txt -------------------------------------------------------------------------------- /DATASET/NL27K/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/NL27K/relations.txt -------------------------------------------------------------------------------- /DATASET/NL27K/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/NL27K/test.txt -------------------------------------------------------------------------------- /DATASET/NL27K/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/NL27K/train.txt -------------------------------------------------------------------------------- /DATASET/NL27K/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/NL27K/valid.txt -------------------------------------------------------------------------------- /DATASET/UFB15K237/entity_id.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/UFB15K237/entity_id.tsv -------------------------------------------------------------------------------- /DATASET/UFB15K237/relation_id.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/UFB15K237/relation_id.tsv -------------------------------------------------------------------------------- /DATASET/UFB15K237/test2id.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/UFB15K237/test2id.tsv -------------------------------------------------------------------------------- /DATASET/UFB15K237/train2id.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/UFB15K237/train2id.tsv -------------------------------------------------------------------------------- /DATASET/UFB15K237/valid2id.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/UFB15K237/valid2id.tsv -------------------------------------------------------------------------------- /DATASET/UWN18RR/entity_id.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/UWN18RR/entity_id.tsv -------------------------------------------------------------------------------- /DATASET/UWN18RR/relation_id.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/UWN18RR/relation_id.tsv -------------------------------------------------------------------------------- /DATASET/UWN18RR/test2id.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/UWN18RR/test2id.tsv -------------------------------------------------------------------------------- /DATASET/UWN18RR/train2id.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/UWN18RR/train2id.tsv -------------------------------------------------------------------------------- /DATASET/UWN18RR/valid2id.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/UWN18RR/valid2id.tsv -------------------------------------------------------------------------------- /DATASET/confidece_analyse.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/confidece_analyse.ipynb -------------------------------------------------------------------------------- /DATASET/histogram.eps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/DATASET/histogram.eps -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/README.md -------------------------------------------------------------------------------- /bert-base-uncased/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/bert-base-uncased/config.json -------------------------------------------------------------------------------- /bert-base-uncased/pytorch_model.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/bert-base-uncased/pytorch_model.bin -------------------------------------------------------------------------------- /bert-base-uncased/tokenizer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/bert-base-uncased/tokenizer.json -------------------------------------------------------------------------------- /bert-base-uncased/tokenizer_config.json: -------------------------------------------------------------------------------- 1 | { 2 | "do_lower_case": true 3 | } 4 | -------------------------------------------------------------------------------- /bert-base-uncased/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/bert-base-uncased/vocab.txt -------------------------------------------------------------------------------- /confidence_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/confidence_prediction.py -------------------------------------------------------------------------------- /decode_rules/dataset/CN15K/cn15k_path_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/decode_rules/dataset/CN15K/cn15k_path_test.txt -------------------------------------------------------------------------------- /decode_rules/dataset/CN15K/cn15k_path_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/decode_rules/dataset/CN15K/cn15k_path_train.txt -------------------------------------------------------------------------------- /decode_rules/dataset/CN15K/cn15k_path_valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/decode_rules/dataset/CN15K/cn15k_path_valid.txt -------------------------------------------------------------------------------- /decode_rules/dataset/NL27K/nl27k_path_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/decode_rules/dataset/NL27K/nl27k_path_test.txt -------------------------------------------------------------------------------- /decode_rules/dataset/NL27K/nl27k_path_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/decode_rules/dataset/NL27K/nl27k_path_train.txt -------------------------------------------------------------------------------- /decode_rules/dataset/NL27K/nl27k_path_valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/decode_rules/dataset/NL27K/nl27k_path_valid.txt -------------------------------------------------------------------------------- /transformer/Layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/transformer/Layers.py -------------------------------------------------------------------------------- /transformer/Models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/transformer/Models.py -------------------------------------------------------------------------------- /transformer/Modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/transformer/Modules.py -------------------------------------------------------------------------------- /transformer/Optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/transformer/Optim.py -------------------------------------------------------------------------------- /transformer/SubLayers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/transformer/SubLayers.py -------------------------------------------------------------------------------- /transformer/Translator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/transformer/Translator.py -------------------------------------------------------------------------------- /transformer/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/transformer/dataset.py -------------------------------------------------------------------------------- /translate_decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/translate_decode.py -------------------------------------------------------------------------------- /translate_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/seucoin/UKRM/HEAD/translate_train.py --------------------------------------------------------------------------------