BPRLoss

class dhg.nn.BPRLoss(*args, **kwargs)[source]

Bases: torch.nn.Module

This criterion computes the Bayesian Personalized Ranking (BPR) loss between the positive scores and the negative scores.

Parameters

alpha (float, optional) – The weight for the positive scores in the BPR loss. Defaults to 1.0.
beta (float, optional) – The weight for the negative scores in the BPR loss. Defaults to 1.0.
activation (str, optional) – The activation function to use can be one of "sigmoid_then_log", "softplus". Defaults to "sigmoid_then_log".

forward(pos_scores, neg_scores)[source]

The forward function of BPRLoss.

Parameters