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

digraph inheritancec005d7e5ca { bgcolor=transparent; rankdir=LR; size="8.0, 12.0"; "ABC" [fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",tooltip="Helper class that provides a standard way to create an ABC using"]; "BasesDecomposition" [URL="../../api/pykeen.nn.message_passing.BasesDecomposition.html#pykeen.nn.message_passing.BasesDecomposition",fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",target="_top",tooltip="Represent relation-weights as a linear combination of base transformation matrices."]; "Decomposition" -> "BasesDecomposition" [arrowsize=0.5,style="setlinewidth(0.5)"]; "BlockDecomposition" [URL="../../api/pykeen.nn.message_passing.BlockDecomposition.html#pykeen.nn.message_passing.BlockDecomposition",fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",target="_top",tooltip="Represent relation-specific weight matrices via block-diagonal matrices."]; "Decomposition" -> "BlockDecomposition" [arrowsize=0.5,style="setlinewidth(0.5)"]; "Decomposition" [URL="../../api/pykeen.nn.message_passing.Decomposition.html#pykeen.nn.message_passing.Decomposition",fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",target="_top",tooltip="Base module for relation-specific message passing."]; "Module" -> "Decomposition" [arrowsize=0.5,style="setlinewidth(0.5)"]; "ExtraReprMixin" -> "Decomposition" [arrowsize=0.5,style="setlinewidth(0.5)"]; "ABC" -> "Decomposition" [arrowsize=0.5,style="setlinewidth(0.5)"]; "ExtraReprMixin" [URL="../utils.html#pykeen.utils.ExtraReprMixin",fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",target="_top",tooltip="A mixin for modules with hierarchical `extra_repr`."]; "Module" [fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",tooltip="Base class for all neural network modules."]; "RGCNLayer" [URL="../../api/pykeen.nn.message_passing.RGCNLayer.html#pykeen.nn.message_passing.RGCNLayer",fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",target="_top",tooltip="An RGCN layer from [schlichtkrull2018]_ updated to match the official implementation."]; "Module" -> "RGCNLayer" [arrowsize=0.5,style="setlinewidth(0.5)"]; "RGCNRepresentation" [URL="../../api/pykeen.nn.message_passing.RGCNRepresentation.html#pykeen.nn.message_passing.RGCNRepresentation",fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",target="_top",tooltip="Entity representations enriched by R-GCN."]; "Representation" -> "RGCNRepresentation" [arrowsize=0.5,style="setlinewidth(0.5)"]; "Representation" [URL="../../api/pykeen.nn.representation.Representation.html#pykeen.nn.representation.Representation",fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",target="_top",tooltip="A base class for obtaining representations for entities/relations."]; "Module" -> "Representation" [arrowsize=0.5,style="setlinewidth(0.5)"]; "ExtraReprMixin" -> "Representation" [arrowsize=0.5,style="setlinewidth(0.5)"]; "ABC" -> "Representation" [arrowsize=0.5,style="setlinewidth(0.5)"]; }