def get_precision_recall()

in privacy_lint/attack_results.py [0:0]

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

    def get_precision_recall(self) -> Tuple[torch.Tensor, torch.Tensor]:
        """
        Computes precision and recall, useful for plotting PR curves and
        computing mAP.
        """

        labels_ordered, _ = self._get_scores_and_labels_ordered()

        true_positives = torch.cumsum(labels_ordered, 0)
        precision = true_positives / torch.arange(1, labels_ordered.shape[0] + 1)
        recall = true_positives / labels_ordered.sum()

        return precision, recall