Predict head/tail in new PyTorch version

thunlp / OpenKE

An Open-Source Package for Knowledge Embedding (KE)

3.83k stars 985 forks source link

Predict head/tail in new PyTorch version #276

Closed akastrin closed 3 years ago

akastrin commented 4 years ago

I go through the example in train_transe_FB15K237.py. I wonder how to get a prediction for the head or tail entity. I get info on that for older OpenKE versions but not for the new PyTorch version.

Thanks, Andrej

LiuYuemei111 commented 3 years ago

Please refer to this https://github.com/thunlp/OpenKE/issues/319#issuecomment-925547973