def get_loss()

in rbr_weight_fitter.py [0:0]

• 6 lines of code
• 1 McCabe index (conditional complexity)

    def get_loss(self, idxs1, idxs2, margins, base_rewards, vals):
        total_scores = base_rewards + vals @ self.weights
        per_example_loss = F.relu(((self.margin * margins) + total_scores[idxs2]) - total_scores[idxs1])
        frac_clipped = 1 - (per_example_loss > 0).float().mean().item()
        loss = per_example_loss.mean()
        return loss, frac_clipped