├── .gitignore ├── LICENSE ├── README.md ├── examples ├── DistMult │ └── DistMult-FB15k.sh ├── RESCAL │ └── RESCAL-FB15k.sh ├── TransE │ └── TransE-FB15k.sh ├── TransH │ └── TransH-FB15k.sh └── TransR │ └── TransR-FB15k.sh ├── images └── structure.png ├── krl ├── __init__.py ├── base_model.py ├── config.py ├── dataset │ ├── __init__.py │ ├── instance │ │ ├── __init__.py │ │ ├── huggingface_krl_datasets_conf.json │ │ ├── index.py │ │ └── utils.py │ └── krl_dataset.py ├── evaluator.py ├── lit_model │ ├── TransXLitModel.py │ └── __init__.py ├── metric.py ├── metric_fomatter.py ├── models │ ├── DistMult.py │ ├── RESCAL.py │ ├── TransD.py │ ├── TransE.py │ ├── TransH.py │ ├── TransR.py │ └── __init__.py ├── negative_sampler.py ├── serializer.py ├── storage.py ├── trainer.py ├── typer_apps │ ├── __init__.py │ ├── distmult.py │ ├── rescal.py │ ├── transe.py │ ├── transh.py │ └── transr.py └── utils │ ├── __init__.py │ ├── data.py │ ├── device.py │ ├── logs_dir.py │ ├── optim.py │ └── seed.py ├── requirements.txt ├── test.ipynb ├── transe.ipynb └── typer_app.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/README.md -------------------------------------------------------------------------------- /examples/DistMult/DistMult-FB15k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/examples/DistMult/DistMult-FB15k.sh -------------------------------------------------------------------------------- /examples/RESCAL/RESCAL-FB15k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/examples/RESCAL/RESCAL-FB15k.sh -------------------------------------------------------------------------------- /examples/TransE/TransE-FB15k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/examples/TransE/TransE-FB15k.sh -------------------------------------------------------------------------------- /examples/TransH/TransH-FB15k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/examples/TransH/TransH-FB15k.sh -------------------------------------------------------------------------------- /examples/TransR/TransR-FB15k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/examples/TransR/TransR-FB15k.sh -------------------------------------------------------------------------------- /images/structure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/images/structure.png -------------------------------------------------------------------------------- /krl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /krl/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/base_model.py -------------------------------------------------------------------------------- /krl/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/config.py -------------------------------------------------------------------------------- /krl/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/dataset/__init__.py -------------------------------------------------------------------------------- /krl/dataset/instance/__init__.py: -------------------------------------------------------------------------------- 1 | from .index import load_krl_dataset 2 | -------------------------------------------------------------------------------- /krl/dataset/instance/huggingface_krl_datasets_conf.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/dataset/instance/huggingface_krl_datasets_conf.json -------------------------------------------------------------------------------- /krl/dataset/instance/index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/dataset/instance/index.py -------------------------------------------------------------------------------- /krl/dataset/instance/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/dataset/instance/utils.py -------------------------------------------------------------------------------- /krl/dataset/krl_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/dataset/krl_dataset.py -------------------------------------------------------------------------------- /krl/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/evaluator.py -------------------------------------------------------------------------------- /krl/lit_model/TransXLitModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/lit_model/TransXLitModel.py -------------------------------------------------------------------------------- /krl/lit_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/lit_model/__init__.py -------------------------------------------------------------------------------- /krl/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/metric.py -------------------------------------------------------------------------------- /krl/metric_fomatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/metric_fomatter.py -------------------------------------------------------------------------------- /krl/models/DistMult.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/models/DistMult.py -------------------------------------------------------------------------------- /krl/models/RESCAL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/models/RESCAL.py -------------------------------------------------------------------------------- /krl/models/TransD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/models/TransD.py -------------------------------------------------------------------------------- /krl/models/TransE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/models/TransE.py -------------------------------------------------------------------------------- /krl/models/TransH.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/models/TransH.py -------------------------------------------------------------------------------- /krl/models/TransR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/models/TransR.py -------------------------------------------------------------------------------- /krl/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /krl/negative_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/negative_sampler.py -------------------------------------------------------------------------------- /krl/serializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/serializer.py -------------------------------------------------------------------------------- /krl/storage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/storage.py -------------------------------------------------------------------------------- /krl/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/trainer.py -------------------------------------------------------------------------------- /krl/typer_apps/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /krl/typer_apps/distmult.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/typer_apps/distmult.py -------------------------------------------------------------------------------- /krl/typer_apps/rescal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/typer_apps/rescal.py -------------------------------------------------------------------------------- /krl/typer_apps/transe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/typer_apps/transe.py -------------------------------------------------------------------------------- /krl/typer_apps/transh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/typer_apps/transh.py -------------------------------------------------------------------------------- /krl/typer_apps/transr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/typer_apps/transr.py -------------------------------------------------------------------------------- /krl/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/utils/__init__.py -------------------------------------------------------------------------------- /krl/utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/utils/data.py -------------------------------------------------------------------------------- /krl/utils/device.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/utils/device.py -------------------------------------------------------------------------------- /krl/utils/logs_dir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/utils/logs_dir.py -------------------------------------------------------------------------------- /krl/utils/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/utils/optim.py -------------------------------------------------------------------------------- /krl/utils/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/krl/utils/seed.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | pytorch 2 | typer[all] 3 | pydantic 4 | loguru 5 | -------------------------------------------------------------------------------- /test.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/test.ipynb -------------------------------------------------------------------------------- /transe.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/transe.ipynb -------------------------------------------------------------------------------- /typer_app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yubinCloud/kg2vec/HEAD/typer_app.py --------------------------------------------------------------------------------