PairwiseLogisticLoss
- class PairwiseLogisticLoss(reduction: str = 'mean')[源代码]
-
成对逻辑损失。
\[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__()
属性摘要
优化损失超参数的默认策略
属性文档