SetwiseLoss

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

Bases: pykeen.losses.Loss

Setwise loss functions compare the scores of several triples.

Initialize the loss.

Parameters

reduction (str) – the reduction, cf. _Loss.__init__