def __init__()

in privacy_lint/attack_results.py [0:0]

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

    def __init__(self, scores_train: torch.Tensor, scores_test: torch.Tensor):
        """
        Given an attack that outputs scores for train and test samples, computes
        attack results.

        Notes:
            - Score should be high for a train sample and low for an test sample:
              typically, -loss is a good score.
        """
        assert scores_train.ndim == scores_test.ndim == 1

        self.scores_train, self.scores_test = scores_train, scores_test