Stateful Interaction Modules

Stateful interaction functions.

Classes

Interaction()

Base class for interaction functions.

FunctionalInteraction()

Base class for interaction functions.

NormBasedInteraction(p[, power_norm])

Norm-based interactions use a (powered) \(p\)-norm in their scoring function.

MonotonicAffineTransformationInteraction(base)

An adapter of interaction functions which adds a scalar (trainable) monotonic affine transformation of the score.

AutoSFInteraction(coefficients)

An implementation of the AutoSF interaction as described by [zhang2020].

BoxEInteraction([tanh_map, p, power_norm])

An implementation of the BoxE interaction from [abboud2020].

ComplExInteraction()

A module wrapper for the stateless ComplEx interaction function.

ConvEInteraction([input_channels, ...])

A stateful module for the ConvE interaction function.

ConvKBInteraction([hidden_dropout_rate, ...])

A stateful module for the ConvKB interaction function.

CPInteraction()

An implementation of the CP interaction as described [lacroix2018] (originally from [hitchcock1927]).

CrossEInteraction([embedding_dim, ...])

A module wrapper for the CrossE interaction function.

DistMAInteraction()

A module wrapper for the stateless DistMA interaction function.

DistMultInteraction()

A module wrapper for the stateless DistMult interaction function.

ERMLPEInteraction([embedding_dim, ...])

A stateful module for the ER-MLP (E) interaction function.

ERMLPInteraction(embedding_dim[, hidden_dim])

A stateful module for the ER-MLP interaction.

HolEInteraction()

A module wrapper for the stateless HolE interaction function.

KG2EInteraction([similarity, exact])

A stateful module for the KG2E interaction function.

LineaREInteraction(p[, power_norm])

The LineaRE interaction described by [peng2020].

MultiLinearTuckerInteraction([head_dim, ...])

An implementation of the original (multi-linear) TuckER interaction as described [tucker1966].

MuREInteraction(p[, power_norm])

A stateful module for the MuRE interaction function from [balazevic2019b].

NTNInteraction([activation, activation_kwargs])

A stateful module for the NTN interaction function.

PairREInteraction(p[, power_norm])

A stateful module for the PairRE interaction function.

ProjEInteraction([embedding_dim, ...])

A stateful module for the ProjE interaction function.

QuatEInteraction()

A module wrapper for the QuatE interaction function.

RESCALInteraction()

A module wrapper for the stateless RESCAL interaction function.

RotatEInteraction()

A module wrapper for the stateless RotatE interaction function.

SEInteraction(p[, power_norm])

A stateful module for the Structured Embedding (SE) interaction function.

SimplEInteraction([clamp_score])

A module wrapper for the SimplE interaction function.

TorusEInteraction([p, power_norm])

A stateful module for the TorusE interaction function.

TransDInteraction([p, power_norm])

A stateful module for the TransD interaction function.

TransEInteraction(p[, power_norm])

A stateful module for the TransE interaction function.

TransFInteraction()

A stateless module for the TransF interaction function.

TransformerInteraction([input_dim, ...])

Transformer-based interaction, as described in [galkin2020].

TransHInteraction(p[, power_norm])

A stateful module for the TransH interaction function.

TransRInteraction(p[, power_norm])

A stateful module for the TransR interaction function.

TripleREInteraction([u, p, power_norm])

A stateful module for the TripleRE interaction function from [yu2021].

TuckerInteraction([embedding_dim, ...])

A stateful module for the stateless Tucker interaction function.

UMInteraction(p[, power_norm])

A stateful module for the UnstructuredModel interaction function.

Class Inheritance Diagram

Inheritance diagram of pykeen.nn.modules.Interaction, pykeen.nn.modules.FunctionalInteraction, pykeen.nn.modules.NormBasedInteraction, pykeen.nn.modules.MonotonicAffineTransformationInteraction, pykeen.nn.modules.AutoSFInteraction, pykeen.nn.modules.BoxEInteraction, pykeen.nn.modules.ComplExInteraction, pykeen.nn.modules.ConvEInteraction, pykeen.nn.modules.ConvKBInteraction, pykeen.nn.modules.CPInteraction, pykeen.nn.modules.CrossEInteraction, pykeen.nn.modules.DistMAInteraction, pykeen.nn.modules.DistMultInteraction, pykeen.nn.modules.ERMLPEInteraction, pykeen.nn.modules.ERMLPInteraction, pykeen.nn.modules.HolEInteraction, pykeen.nn.modules.KG2EInteraction, pykeen.nn.modules.LineaREInteraction, pykeen.nn.modules.MultiLinearTuckerInteraction, pykeen.nn.modules.MuREInteraction, pykeen.nn.modules.NTNInteraction, pykeen.nn.modules.PairREInteraction, pykeen.nn.modules.ProjEInteraction, pykeen.nn.modules.QuatEInteraction, pykeen.nn.modules.RESCALInteraction, pykeen.nn.modules.RotatEInteraction, pykeen.nn.modules.SEInteraction, pykeen.nn.modules.SimplEInteraction, pykeen.nn.modules.TorusEInteraction, pykeen.nn.modules.TransDInteraction, pykeen.nn.modules.TransEInteraction, pykeen.nn.modules.TransFInteraction, pykeen.nn.modules.TransformerInteraction, pykeen.nn.modules.TransHInteraction, pykeen.nn.modules.TransRInteraction, pykeen.nn.modules.TripleREInteraction, pykeen.nn.modules.TuckerInteraction, pykeen.nn.modules.UMInteraction