Weighting

Various edge weighting implementations for R-GCN.

Classes

EdgeWeighting(**kwargs)

Base class for edge weightings.

InverseInDegreeEdgeWeighting(**kwargs)

Normalize messages by inverse in-degree.

InverseOutDegreeEdgeWeighting(**kwargs)

Normalize messages by inverse out-degree.

SymmetricEdgeWeighting(**kwargs)

Normalize messages by product of inverse sqrt of in-degree and out-degree.

AttentionEdgeWeighting(message_dim[, ...])

Message weighting by attention.

Class Inheritance Diagram

Inheritance diagram of pykeen.nn.weighting.EdgeWeighting, pykeen.nn.weighting.InverseInDegreeEdgeWeighting, pykeen.nn.weighting.InverseOutDegreeEdgeWeighting, pykeen.nn.weighting.SymmetricEdgeWeighting, pykeen.nn.weighting.AttentionEdgeWeighting