MSELoss
- class MSELoss(reduction: str = 'mean')[source]
Bases:
PointwiseLoss
The mean squared error loss.
Note
The related
torch
module istorch.nn.MSELoss
, 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
Attributes Summary
synonyms of this loss
Methods Summary
forward
(x, target[, weight])Calculate the point-wise loss.
Attributes Documentation
- synonyms: ClassVar[set[str] | None] = {'Mean Square Error Loss', 'Mean Squared Error Loss'}
synonyms of this loss
Methods Documentation