├── .gitignore ├── README.md ├── datasets ├── README.md ├── family │ ├── all.txt │ ├── entities.txt │ ├── facts.txt │ ├── relations.txt │ ├── test.txt │ ├── train.txt │ ├── truths.pckl │ └── valid.txt ├── fb15k-237 │ ├── entities.txt │ ├── facts.txt │ ├── relations.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── kinship │ ├── entities.txt │ ├── facts.txt │ ├── relations.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── umls │ ├── entities.txt │ ├── facts.txt │ ├── relations.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt └── wn-18 │ ├── entities.txt │ ├── facts.txt │ ├── relations.txt │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── eval ├── collect_all_facts.sh ├── evaluate.py ├── evaluate.sh └── get_truths.py └── src ├── data.py ├── experiment.py ├── main.py └── model.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/README.md -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/README.md -------------------------------------------------------------------------------- /datasets/family/all.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/family/all.txt -------------------------------------------------------------------------------- /datasets/family/entities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/family/entities.txt -------------------------------------------------------------------------------- /datasets/family/facts.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/family/facts.txt -------------------------------------------------------------------------------- /datasets/family/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/family/relations.txt -------------------------------------------------------------------------------- /datasets/family/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/family/test.txt -------------------------------------------------------------------------------- /datasets/family/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/family/train.txt -------------------------------------------------------------------------------- /datasets/family/truths.pckl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/family/truths.pckl -------------------------------------------------------------------------------- /datasets/family/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/family/valid.txt -------------------------------------------------------------------------------- /datasets/fb15k-237/entities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/fb15k-237/entities.txt -------------------------------------------------------------------------------- /datasets/fb15k-237/facts.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/fb15k-237/facts.txt -------------------------------------------------------------------------------- /datasets/fb15k-237/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/fb15k-237/relations.txt -------------------------------------------------------------------------------- /datasets/fb15k-237/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/fb15k-237/test.txt -------------------------------------------------------------------------------- /datasets/fb15k-237/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/fb15k-237/train.txt -------------------------------------------------------------------------------- /datasets/fb15k-237/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/fb15k-237/valid.txt -------------------------------------------------------------------------------- /datasets/kinship/entities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/kinship/entities.txt -------------------------------------------------------------------------------- /datasets/kinship/facts.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/kinship/facts.txt -------------------------------------------------------------------------------- /datasets/kinship/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/kinship/relations.txt -------------------------------------------------------------------------------- /datasets/kinship/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/kinship/test.txt -------------------------------------------------------------------------------- /datasets/kinship/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/kinship/train.txt -------------------------------------------------------------------------------- /datasets/kinship/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/kinship/valid.txt -------------------------------------------------------------------------------- /datasets/umls/entities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/umls/entities.txt -------------------------------------------------------------------------------- /datasets/umls/facts.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/umls/facts.txt -------------------------------------------------------------------------------- /datasets/umls/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/umls/relations.txt -------------------------------------------------------------------------------- /datasets/umls/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/umls/test.txt -------------------------------------------------------------------------------- /datasets/umls/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/umls/train.txt -------------------------------------------------------------------------------- /datasets/umls/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/umls/valid.txt -------------------------------------------------------------------------------- /datasets/wn-18/entities.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/wn-18/entities.txt -------------------------------------------------------------------------------- /datasets/wn-18/facts.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/wn-18/facts.txt -------------------------------------------------------------------------------- /datasets/wn-18/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/wn-18/relations.txt -------------------------------------------------------------------------------- /datasets/wn-18/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/wn-18/test.txt -------------------------------------------------------------------------------- /datasets/wn-18/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/wn-18/train.txt -------------------------------------------------------------------------------- /datasets/wn-18/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/datasets/wn-18/valid.txt -------------------------------------------------------------------------------- /eval/collect_all_facts.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/eval/collect_all_facts.sh -------------------------------------------------------------------------------- /eval/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/eval/evaluate.py -------------------------------------------------------------------------------- /eval/evaluate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/eval/evaluate.sh -------------------------------------------------------------------------------- /eval/get_truths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/eval/get_truths.py -------------------------------------------------------------------------------- /src/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/src/data.py -------------------------------------------------------------------------------- /src/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/src/experiment.py -------------------------------------------------------------------------------- /src/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/src/main.py -------------------------------------------------------------------------------- /src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/alisadeghian/DRUM/HEAD/src/model.py --------------------------------------------------------------------------------