def _compute_scaled_loss()

in automl21/scs_neural/solver/neural_scs_batched.py [0:0]

• 14 lines of code
• 7 McCabe index (conditional complexity)

    def _compute_scaled_loss(self, u, fp_u, include_bad_tau=False):
        if self.use_unscaled_loss:
            u, _, scaled_u = self._remove_tau_scaling_from_iterate(u, include_bad_tau)
            fp_u, _, scaled_fp_u = self._remove_tau_scaling_from_iterate(fp_u, include_bad_tau)
            if not scaled_u or not scaled_fp_u:
                return None
        else:
            tau = u[:, -1]
            fp_tau = u[:, -1]
            if not include_bad_tau:
                if (tau <= 0).any() or (fp_tau <= 0).any():
                    return None
        diff_u = (fp_u - u).norm(dim=1)
        return diff_u