SetwiseLoss

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

Bases: Loss

Setwise loss functions compare the scores of several triples.

Initialize the loss.

Parameters

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