TextRepresentation

class TextRepresentation(labels, max_id=None, shape=None, encoder=None, encoder_kwargs=None, **kwargs)

Bases: Representation

Textual representations using a text encoder on labels.

Example Usage:

Entity representations are obtained by encoding the labels with a Transformer model. The transformer model becomes part of the KGE model, and its parameters are trained jointly.

from pykeen.datasets import get_dataset
from pykeen.nn.representation import TextRepresentation
from pykeen.models import ERModel

dataset = get_dataset(dataset="nations")
entity_representations = TextRepresentation.from_dataset(
    triples_factory=dataset,
    encoder="transformer",
)
model = ERModel(
    interaction="ermlp",
    entity_representations=entity_representations,
    relation_representations_kwargs=dict(shape=entity_representations.shape),
)

Initialize the representation.

Parameters

• labels (Sequence[str]) – the labels

• max_id (Optional[int]) – the number of representations. If provided, has to match the number of labels

• shape (Union[int, Sequence[int], None]) – The shape of an individual representation.

• encoder (Union[str, TextEncoder, Type[TextEncoder], None]) – the text encoder, or a hint thereof

• encoder_kwargs (Optional[Mapping[str, Any]]) – keyword-based parameters used to instantiate the text encoder

• kwargs – additional keyword-based parameters passed to Representation.__init__()

Raises

ValueError – if the max_id does not match

Methods Summary

from_dataset(dataset, **kwargs)	Prepare text representation with labels from a dataset.
from_triples_factory(triples_factory[, ...])	Prepare a text representations with labels from a triples factory.

Methods Documentation

classmethod from_dataset(dataset, **kwargs)

Prepare text representation with labels from a dataset.

Parameters

• dataset (Dataset) – the dataset

• kwargs – additional keyword-based parameters passed to TextRepresentation.from_triples_factory()

Return type

ForwardRef

Returns

a text representation from the dataset

Raises

TypeError – if the dataset’s triples factory does not provide labels

classmethod from_triples_factory(triples_factory, for_entities=True, **kwargs)

Prepare a text representations with labels from a triples factory.

Parameters

• triples_factory (TriplesFactory) – the triples factory

• for_entities (bool) – whether to create the initializer for entities (or relations)

• kwargs – additional keyword-based arguments passed to TextRepresentation.__init__()

Return type

ForwardRef

Returns

a text representation from the triples factory