Initialization

Embedding weight initialization routines.

Functions

xavier_uniform_(tensor[, gain])

Initialize weights of the tensor similarly to Glorot/Xavier initialization.

xavier_normal_(tensor[, gain])

Initialize weights of the tensor similarly to Glorot/Xavier initialization.

init_phases(x)

Generate random phases between 0 and \(2\pi\).

Classes

PretrainedInitializer(tensor)

Initialize tensor with pretrained weights.

LabelBasedInitializer(labels[, encoder, ...])

An initializer using pretrained models from the transformers library to encode labels.

WeisfeilerLehmanInitializer(*[, ...])

An initializer based on an encoding of categorical colors from the Weisfeiler-Lehman algorithm.

RandomWalkPositionalEncodingInitializer(*[, ...])

Initialize nodes via random-walk positional encoding.

Class Inheritance Diagram

digraph inheritancee775b8f2ef { bgcolor=transparent; rankdir=LR; size="8.0, 12.0"; "LabelBasedInitializer" [URL="../../api/pykeen.nn.init.LabelBasedInitializer.html#pykeen.nn.init.LabelBasedInitializer",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 initializer using pretrained models from the `transformers` library to encode labels."]; "PretrainedInitializer" -> "LabelBasedInitializer" [arrowsize=0.5,style="setlinewidth(0.5)"]; "PretrainedInitializer" [URL="../../api/pykeen.nn.init.PretrainedInitializer.html#pykeen.nn.init.PretrainedInitializer",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="Initialize tensor with pretrained weights."]; "RandomWalkPositionalEncodingInitializer" [URL="../../api/pykeen.nn.init.RandomWalkPositionalEncodingInitializer.html#pykeen.nn.init.RandomWalkPositionalEncodingInitializer",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="Initialize nodes via random-walk positional encoding."]; "PretrainedInitializer" -> "RandomWalkPositionalEncodingInitializer" [arrowsize=0.5,style="setlinewidth(0.5)"]; "WeisfeilerLehmanInitializer" [URL="../../api/pykeen.nn.init.WeisfeilerLehmanInitializer.html#pykeen.nn.init.WeisfeilerLehmanInitializer",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 initializer based on an encoding of categorical colors from the Weisfeiler-Lehman algorithm."]; "PretrainedInitializer" -> "WeisfeilerLehmanInitializer" [arrowsize=0.5,style="setlinewidth(0.5)"]; }