SetwiseLoss

class SetwiseLoss(reduction: str = 'mean')[source]

Bases: Loss

Setwise loss functions compare the scores of several triples.

Initialize the loss.

Parameters:

reduction (str) – the reduction, cf. pykeen.nn.modules._Loss