├── .gitignore ├── README.md ├── datasets ├── __init__.py ├── download.sh ├── kg_dataset.py └── process.py ├── examples ├── train_ComplEx_WN18RR_32.sh ├── train_ComplEx_WN18RR_500.sh ├── train_RotE_WN18RR_32.sh ├── train_RotE_WN18RR_500.sh ├── train_RotH_FB237_32.sh ├── train_RotH_WN18RR_32.sh ├── train_RotH_WN18RR_500.sh └── train_RotH_YAGO3-10_32.sh ├── models ├── __init__.py ├── base.py ├── complex.py ├── euclidean.py └── hyperbolic.py ├── optimizers ├── __init__.py ├── kg_optimizer.py └── regularizers.py ├── requirements.txt ├── run.py ├── set_env.sh ├── test.py └── utils ├── __init__.py ├── euclidean.py ├── hyperbolic.py └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | data/ 2 | logs/ 3 | */__pycache__ 4 | hyp_kg_env/ 5 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/README.md -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/datasets/download.sh -------------------------------------------------------------------------------- /datasets/kg_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/datasets/kg_dataset.py -------------------------------------------------------------------------------- /datasets/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/datasets/process.py -------------------------------------------------------------------------------- /examples/train_ComplEx_WN18RR_32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/examples/train_ComplEx_WN18RR_32.sh -------------------------------------------------------------------------------- /examples/train_ComplEx_WN18RR_500.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/examples/train_ComplEx_WN18RR_500.sh -------------------------------------------------------------------------------- /examples/train_RotE_WN18RR_32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/examples/train_RotE_WN18RR_32.sh -------------------------------------------------------------------------------- /examples/train_RotE_WN18RR_500.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/examples/train_RotE_WN18RR_500.sh -------------------------------------------------------------------------------- /examples/train_RotH_FB237_32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/examples/train_RotH_FB237_32.sh -------------------------------------------------------------------------------- /examples/train_RotH_WN18RR_32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/examples/train_RotH_WN18RR_32.sh -------------------------------------------------------------------------------- /examples/train_RotH_WN18RR_500.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/examples/train_RotH_WN18RR_500.sh -------------------------------------------------------------------------------- /examples/train_RotH_YAGO3-10_32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/examples/train_RotH_YAGO3-10_32.sh -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/models/base.py -------------------------------------------------------------------------------- /models/complex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/models/complex.py -------------------------------------------------------------------------------- /models/euclidean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/models/euclidean.py -------------------------------------------------------------------------------- /models/hyperbolic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/models/hyperbolic.py -------------------------------------------------------------------------------- /optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/optimizers/__init__.py -------------------------------------------------------------------------------- /optimizers/kg_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/optimizers/kg_optimizer.py -------------------------------------------------------------------------------- /optimizers/regularizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/optimizers/regularizers.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | torch 3 | tqdm 4 | -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/run.py -------------------------------------------------------------------------------- /set_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/set_env.sh -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/test.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/euclidean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/utils/euclidean.py -------------------------------------------------------------------------------- /utils/hyperbolic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/utils/hyperbolic.py -------------------------------------------------------------------------------- /utils/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HazyResearch/KGEmb/HEAD/utils/train.py --------------------------------------------------------------------------------