Representation

Representation modules.

Classes

Representation(max_id[, shape, normalizer, ...])

A base class for obtaining representations for entities/relations.

Embedding([max_id, num_embeddings, ...])

Trainable embeddings.

LowRankRepresentation(*, max_id, shape[, ...])

Low-rank embedding factorization.

CompGCNLayer(input_dim[, output_dim, ...])

A single layer of the CompGCN model.

CombinedCompGCNRepresentations(*, ...[, ...])

A sequence of CompGCN layers.

PartitionRepresentation(assignment[, shape, ...])

A partition of the indices into different representation modules.

BackfillRepresentation(max_id, base_ids[, ...])

A variant of a partition representation that is easily applicable to a single base representation.

SingleCompGCNRepresentation(combined[, ...])

A wrapper around the combined representation module.

SubsetRepresentation(max_id[, base, ...])

A representation module, which only exposes a subset of representations of its base.

CombinedRepresentation(max_id[, shape, ...])

A combined representation.

TensorTrainRepresentation([assignment, ...])

A tensor factorization of representations.

TransformedRepresentation(transformation[, ...])

A (learnable) transformation upon base representations.

TextRepresentation(labels[, max_id, shape, ...])

Textual representations using a text encoder on labels.

CachedTextRepresentation(identifiers, **kwargs)

Textual representations for datasets with identifiers that can be looked up with a TextCache.

WikidataTextRepresentation(identifiers, **kwargs)

Textual representations for datasets grounded in Wikidata.

BiomedicalCURIERepresentation(identifiers, ...)

Textual representations for datasets grounded with biomedical CURIEs.

Class Inheritance Diagram

Inheritance diagram of pykeen.nn.representation.Representation, pykeen.nn.representation.Embedding, pykeen.nn.representation.LowRankRepresentation, pykeen.nn.representation.CompGCNLayer, pykeen.nn.representation.CombinedCompGCNRepresentations, pykeen.nn.representation.PartitionRepresentation, pykeen.nn.representation.BackfillRepresentation, pykeen.nn.representation.SingleCompGCNRepresentation, pykeen.nn.representation.SubsetRepresentation, pykeen.nn.representation.CombinedRepresentation, pykeen.nn.representation.TensorTrainRepresentation, pykeen.nn.representation.TransformedRepresentation, pykeen.nn.representation.TextRepresentation, pykeen.nn.representation.CachedTextRepresentation, pykeen.nn.representation.WikidataTextRepresentation, pykeen.nn.representation.BiomedicalCURIERepresentation