Message Passing

Various decompositions for R-GCN.

Classes

RGCNRepresentation(triples_factory[, ...])

Entity representations enriched by R-GCN.

RGCNLayer(num_relations[, input_dim, ...])

An RGCN layer from [schlichtkrull2018] updated to match the official implementation.

Decomposition(num_relations[, input_dim, ...])

Base module for relation-specific message passing.

BasesDecomposition([num_bases])

Represent relation-weights as a linear combination of base transformation matrices.

BlockDecomposition([num_blocks])

Represent relation-specific weight matrices via block-diagonal matrices.

Class Inheritance Diagram

Inheritance diagram of pykeen.nn.message_passing.RGCNRepresentation, pykeen.nn.message_passing.RGCNLayer, pykeen.nn.message_passing.Decomposition, pykeen.nn.message_passing.BasesDecomposition, pykeen.nn.message_passing.BlockDecomposition