BCEAfterSigmoidLoss
- class BCEAfterSigmoidLoss(reduction: str = 'mean')[source]
Bases:
PointwiseLoss
The numerically unstable version of explicit Sigmoid + BCE loss.
Note
The related
torch
module istorch.nn.BCELoss
, but it can not be used interchangeably in PyKEEN because of the extended functionality implemented in PyKEEN’s loss functions.Initialize the loss.
- Parameters:
reduction (str) – the reduction, cf.
pykeen.nn.modules._Loss
Methods Summary
forward
(x, target[, weight])Calculate the point-wise loss.
Methods Documentation