torchkge-team / torchkge

TorchKGE: Knowledge Graph embedding in Python and PyTorch.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Can there be an example for ConvKB and RotatE

LemonQC opened this issue · comments

Hello @SakerLiu

RotatE is not yet implemented in TorchKGE but feel free to submit a merge request.

When it comes to ConvKB, the provided examples can simply be adapted simply by changing the model constructor to ConvKB (and provided the proper parameters of course to the constructor, namely n_filters).