Stateful Interaction Modules

Stateful interaction functions.

Classes

Interaction(*args, **kwargs)	Base class for interaction functions.
FunctionalInteraction(*args, **kwargs)	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, *[, ...])	The AutoSF interaction as described by [zhang2020].
BoxEInteraction([tanh_map, p, power_norm])	An implementation of the BoxE interaction from [abboud2020].
ComplExInteraction(*args, **kwargs)	The ComplEx interaction proposed by [trouillon2016].
ConvEInteraction([input_channels, ...])	A stateful module for the ConvE interaction function.
ConvKBInteraction([hidden_dropout_rate, ...])	A stateful module for the ConvKB interaction function.
CPInteraction(*args, **kwargs)	An implementation of the CP interaction as described [lacroix2018] (originally from [hitchcock1927]).
CrossEInteraction([embedding_dim, ...])	A module wrapper for the CrossE interaction function.
DistMAInteraction(*args, **kwargs)	A module wrapper for the stateless DistMA interaction function.
DistMultInteraction(*args, **kwargs)	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(*args, **kwargs)	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(*args, **kwargs)	A module wrapper for the stateless RESCAL interaction function.
RotatEInteraction(*args, **kwargs)	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(*args, **kwargs)	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