RotatE

RotatE scoring function.

Attributes

• name

Examples

>>> from ckb import models
>>> from ckb import datasets
>>> from ckb import scoring

>>> import torch

>>> _ = torch.manual_seed(42)

>>> dataset = datasets.Semanlink(1)

>>> model = models.DistillBert(
...    entities = dataset.entities,
...    relations = dataset.relations,
...    gamma = 9,
...    device = 'cpu',
...    scoring = scoring.RotatE(),
... )

>>> sample = torch.tensor([[0, 0, 1], [2, 2, 1]])
>>> model(sample)
tensor([[-203.6809],
        [-191.3758]], grad_fn=<ViewBackward>)

>>> sample = torch.tensor([[1, 0, 0], [1, 2, 2]])
>>> model(sample)
tensor([[-204.0743],
        [-192.8306]], grad_fn=<ViewBackward>)

Methods

call

Compute the score of given facts (heads, relations, tails).

Parameters

• head
• relation
• tail
• gamma
• embedding_range
• mode
• kwargs