PairwiseLogisticLoss

class PairwiseLogisticLoss(reduction: str = 'mean')[源代码]

基础类:SoftMarginRankingLoss

成对逻辑损失。

\[L(k, \bar{k}) = \log(1 + \exp(f(\bar{k}) - f(k)))\]

其中 \(k\) 是正三元组,\(\bar{k}\) 是负三元组,\(f\) 是交互函数(例如, pykeen.models.TransE\(f(h,r,t)=-||\mathbf{e}_h+\mathbf{e}_r-\mathbf{e}_t||_p\)), \(g(x)=\log(1 + \exp(x))\) 是 softmax 激活函数。

另请参阅

这种损失等同于pykeen.losses.SoftMarginRankingLoss,其中margin=0。它也与pykeen.losses.MarginRankingLoss密切相关,基于激活函数的选择。

初始化损失。

Parameters:

reduction (str) – 减少,参见 SoftMarginRankingLoss.__init__()

属性摘要

hpo_default

优化损失超参数的默认策略

属性文档

hpo_default: ClassVar[Mapping[str, Any]] = {}

优化损失超参数的默认策略