Sujit-O / pykg2vec

Python library for knowledge graph embedding and representation learning.
MIT License
606 stars 111 forks source link

configure hyperparameters in YAML files and support customer override #159

Closed baxtree closed 4 years ago

baxtree commented 4 years ago

Golden/paper hyperparameters are defined in the pykg2vec/hyperparams folder.

Customers can tweak the values inside YAML files under ./examples/hyperparams or create your own files, which will override the above golden/paper hyperparameters through

$ python train.py -mn TransE -ds [name] -exp True -hpd ./examples/hyperparams