有状态交互模块

有状态交互函数。

Interaction(*args, **kwargs)

交互功能的基类。

FunctionalInteraction(*args, **kwargs)

交互功能的基类。

NormBasedInteraction(p[, power_norm])

基于范数的交互在其评分函数中使用(幂)\(p\)-范数。

MonotonicAffineTransformationInteraction(base)

一个交互函数的适配器,它添加了一个可训练的单调仿射变换的分数。

ClampedInteraction(base[, base_kwargs, ...])

一个适配器,用于将分数限制在最小值或最大值。

DirectionAverageInteraction(base[, base_kwargs])

方向平均交互模块。

AutoSFInteraction(coefficients, *[, ...])

AutoSF交互如[zhang2020]所述。

BoxEInteraction([tanh_map, p, power_norm])

来自[abboud2020]的BoxE交互。

ComplExInteraction(*args, **kwargs)

[trouillon2016]提出的ComplEx交互。

ConvEInteraction([input_channels, ...])

有状态的ConvE交互函数。

ConvKBInteraction([hidden_dropout_rate, ...])

有状态的ConvKB交互函数。

CPInteraction(*args, **kwargs)

规范张量分解交互,如[lacroix2018]所述(最初来自[hitchcock1927])。

CrossEInteraction(embedding_dim, ...)

CrossE的有状态交互函数。

DistMAInteraction(*args, **kwargs)

来自[shi2019]的无状态DistMA交互函数。

DistMultInteraction(*args, **kwargs)

无状态的DistMult交互函数。

ERMLPEInteraction([embedding_dim, ...])

有状态的ER-MLP(E)交互函数。

ERMLPInteraction(embedding_dim, hidden_dim, ...)

ER-MLP状态交互函数。

HolEInteraction(*args, **kwargs)

无状态的HolE交互函数。

KG2EInteraction([similarity, similarity_kwargs])

无状态的KG2E交互函数。

LineaREInteraction(p[, power_norm])

[peng2020]描述的LineaRE交互。

MultiLinearTuckerInteraction([head_dim, ...])

原始(多线性)TuckER交互的实现,如[tucker1966]所述。

MuREInteraction(p[, power_norm])

基于范数的MuRE交互函数来自[balazevic2019b]

NTNInteraction([activation, activation_kwargs])

无状态的神经张量网络(NTN)交互函数。

PairREInteraction(p[, power_norm])

基于无状态范数的PairRE交互函数。

ProjEInteraction(embedding_dim, ...)

有状态的ProjE交互函数。

QuatEInteraction()

无状态的QuatE交互函数。

RESCALInteraction(*args, **kwargs)

无状态的RESCAL交互函数。

RotatEInteraction(*args, **kwargs)

[sun2019]提出的RotatE交互函数。

SEInteraction(p[, power_norm])

结构化嵌入(SE)交互函数。

SimplEInteraction()

SimplE交互函数。

TorusEInteraction([p, power_norm])

来自[ebisu2018]的TorusE交互函数。

TransDInteraction([p, power_norm])

TransD交互函数。

TransEInteraction(p[, power_norm])

基于无状态范数的TransE交互函数。

TransFInteraction(*args, **kwargs)

基于无状态范数的TransF交互函数。

TransformerInteraction(input_dim, ...)

基于Transformer的交互,如[galkin2020]所述。

TransHInteraction(p[, power_norm])

基于范数的TransH交互函数。

TransRInteraction(p[, power_norm, ...])

基于无状态范数的TransR交互函数。

TripleREInteraction([u, p, power_norm])

TripleRE交互函数来自[yu2021]

TuckERInteraction([embedding_dim, ...])

有状态的TuckER交互函数。

UMInteraction(p[, power_norm])

非结构化模型(UM)交互函数。

变量

interaction_resolver

用于有状态交互函数的解析器

类继承图

Inheritance diagram of pykeen.nn.modules.Interaction, pykeen.nn.modules.FunctionalInteraction, pykeen.nn.modules.NormBasedInteraction, pykeen.nn.modules.MonotonicAffineTransformationInteraction, pykeen.nn.modules.ClampedInteraction, pykeen.nn.modules.DirectionAverageInteraction, 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