├── .gitignore ├── README.md ├── core ├── ComplEx.py ├── DataSampler.py ├── DistMult.py ├── TransE.py └── __init__.py ├── data ├── FB15K237 │ ├── FB15K237.pickle │ ├── README.txt │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── FB15k │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── MiniFB15K237-v1 │ ├── MiniFB15K237-v1.pickle │ ├── README.txt │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── MiniFB15K237-v2 │ ├── MiniFB15K237-v2.pickle │ ├── README.txt │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── MiniFB15K237-v3 │ ├── MiniFB15K237-v3.pickle │ ├── README.txt │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── MiniFB15K237-v4 │ ├── MiniFB15K237-v3.pickle │ ├── MiniFB15K237-v4.pickle │ ├── README.txt │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── MiniFB15K237-v5 │ ├── MiniFB15K237-v3.pickle │ ├── MiniFB15K237-v5.pickle │ ├── README.txt │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── Toy │ ├── Toy.pickle │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── USAir.mat ├── tmp │ ├── FB15K237.pickle │ ├── README.txt │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── tmp.pickle │ ├── train.txt │ └── valid.txt ├── toy_tmp │ ├── Toy.pickle │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── toy_tmp.pickle │ ├── train.txt │ └── valid.txt └── wn18 │ ├── README │ ├── Wordnet3.0-LICENSE │ ├── entities.dict │ ├── relations.dict │ ├── test.txt │ ├── train.txt │ └── valid.txt ├── evaluate.py ├── managers ├── Evaluator.py ├── Trainer.py └── __init__.py ├── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | __pycache__/ 3 | experiments/ 4 | tmp.txt 5 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/README.md -------------------------------------------------------------------------------- /core/ComplEx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/core/ComplEx.py -------------------------------------------------------------------------------- /core/DataSampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/core/DataSampler.py -------------------------------------------------------------------------------- /core/DistMult.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/core/DistMult.py -------------------------------------------------------------------------------- /core/TransE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/core/TransE.py -------------------------------------------------------------------------------- /core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/core/__init__.py -------------------------------------------------------------------------------- /data/FB15K237/FB15K237.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/FB15K237/FB15K237.pickle -------------------------------------------------------------------------------- /data/FB15K237/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/FB15K237/README.txt -------------------------------------------------------------------------------- /data/FB15K237/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/FB15K237/entities.dict -------------------------------------------------------------------------------- /data/FB15K237/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/FB15K237/relations.dict -------------------------------------------------------------------------------- /data/FB15K237/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/FB15K237/test.txt -------------------------------------------------------------------------------- /data/FB15K237/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/FB15K237/train.txt -------------------------------------------------------------------------------- /data/FB15K237/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/FB15K237/valid.txt -------------------------------------------------------------------------------- /data/FB15k/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/FB15k/entities.dict -------------------------------------------------------------------------------- /data/FB15k/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/FB15k/relations.dict -------------------------------------------------------------------------------- /data/FB15k/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/FB15k/test.txt -------------------------------------------------------------------------------- /data/FB15k/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/FB15k/train.txt -------------------------------------------------------------------------------- /data/FB15k/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/FB15k/valid.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v1/MiniFB15K237-v1.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v1/MiniFB15K237-v1.pickle -------------------------------------------------------------------------------- /data/MiniFB15K237-v1/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v1/README.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v1/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v1/entities.dict -------------------------------------------------------------------------------- /data/MiniFB15K237-v1/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v1/relations.dict -------------------------------------------------------------------------------- /data/MiniFB15K237-v1/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v1/test.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v1/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v1/train.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v1/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v1/valid.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v2/MiniFB15K237-v2.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v2/MiniFB15K237-v2.pickle -------------------------------------------------------------------------------- /data/MiniFB15K237-v2/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v2/README.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v2/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v2/entities.dict -------------------------------------------------------------------------------- /data/MiniFB15K237-v2/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v2/relations.dict -------------------------------------------------------------------------------- /data/MiniFB15K237-v2/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v2/test.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v2/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v2/train.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v2/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v2/valid.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v3/MiniFB15K237-v3.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v3/MiniFB15K237-v3.pickle -------------------------------------------------------------------------------- /data/MiniFB15K237-v3/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v3/README.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v3/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v3/entities.dict -------------------------------------------------------------------------------- /data/MiniFB15K237-v3/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v3/relations.dict -------------------------------------------------------------------------------- /data/MiniFB15K237-v3/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v3/test.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v3/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v3/train.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v3/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v3/valid.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v4/MiniFB15K237-v3.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v4/MiniFB15K237-v3.pickle -------------------------------------------------------------------------------- /data/MiniFB15K237-v4/MiniFB15K237-v4.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v4/MiniFB15K237-v4.pickle -------------------------------------------------------------------------------- /data/MiniFB15K237-v4/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v4/README.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v4/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v4/entities.dict -------------------------------------------------------------------------------- /data/MiniFB15K237-v4/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v4/relations.dict -------------------------------------------------------------------------------- /data/MiniFB15K237-v4/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v4/test.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v4/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v4/train.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v4/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v4/valid.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v5/MiniFB15K237-v3.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v5/MiniFB15K237-v3.pickle -------------------------------------------------------------------------------- /data/MiniFB15K237-v5/MiniFB15K237-v5.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v5/MiniFB15K237-v5.pickle -------------------------------------------------------------------------------- /data/MiniFB15K237-v5/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v5/README.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v5/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v5/entities.dict -------------------------------------------------------------------------------- /data/MiniFB15K237-v5/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v5/relations.dict -------------------------------------------------------------------------------- /data/MiniFB15K237-v5/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v5/test.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v5/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v5/train.txt -------------------------------------------------------------------------------- /data/MiniFB15K237-v5/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/MiniFB15K237-v5/valid.txt -------------------------------------------------------------------------------- /data/Toy/Toy.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/Toy/Toy.pickle -------------------------------------------------------------------------------- /data/Toy/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/Toy/entities.dict -------------------------------------------------------------------------------- /data/Toy/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/Toy/relations.dict -------------------------------------------------------------------------------- /data/Toy/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/Toy/test.txt -------------------------------------------------------------------------------- /data/Toy/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/Toy/train.txt -------------------------------------------------------------------------------- /data/Toy/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/Toy/valid.txt -------------------------------------------------------------------------------- /data/USAir.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/USAir.mat -------------------------------------------------------------------------------- /data/tmp/FB15K237.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/tmp/FB15K237.pickle -------------------------------------------------------------------------------- /data/tmp/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/tmp/README.txt -------------------------------------------------------------------------------- /data/tmp/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/tmp/entities.dict -------------------------------------------------------------------------------- /data/tmp/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/tmp/relations.dict -------------------------------------------------------------------------------- /data/tmp/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/tmp/test.txt -------------------------------------------------------------------------------- /data/tmp/tmp.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/tmp/tmp.pickle -------------------------------------------------------------------------------- /data/tmp/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/tmp/train.txt -------------------------------------------------------------------------------- /data/tmp/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/tmp/valid.txt -------------------------------------------------------------------------------- /data/toy_tmp/Toy.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/toy_tmp/Toy.pickle -------------------------------------------------------------------------------- /data/toy_tmp/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/toy_tmp/entities.dict -------------------------------------------------------------------------------- /data/toy_tmp/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/toy_tmp/relations.dict -------------------------------------------------------------------------------- /data/toy_tmp/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/toy_tmp/test.txt -------------------------------------------------------------------------------- /data/toy_tmp/toy_tmp.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/toy_tmp/toy_tmp.pickle -------------------------------------------------------------------------------- /data/toy_tmp/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/toy_tmp/train.txt -------------------------------------------------------------------------------- /data/toy_tmp/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/toy_tmp/valid.txt -------------------------------------------------------------------------------- /data/wn18/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/wn18/README -------------------------------------------------------------------------------- /data/wn18/Wordnet3.0-LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/wn18/Wordnet3.0-LICENSE -------------------------------------------------------------------------------- /data/wn18/entities.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/wn18/entities.dict -------------------------------------------------------------------------------- /data/wn18/relations.dict: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/wn18/relations.dict -------------------------------------------------------------------------------- /data/wn18/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/wn18/test.txt -------------------------------------------------------------------------------- /data/wn18/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/wn18/train.txt -------------------------------------------------------------------------------- /data/wn18/valid.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/data/wn18/valid.txt -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/evaluate.py -------------------------------------------------------------------------------- /managers/Evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/managers/Evaluator.py -------------------------------------------------------------------------------- /managers/Trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/managers/Trainer.py -------------------------------------------------------------------------------- /managers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/managers/__init__.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kkteru/node-embeddings/HEAD/utils.py --------------------------------------------------------------------------------