def create_or_retrieve_label()

in src/features/feature_variables_dynamic.py [0:0]

• 11 lines of code
• 4 McCabe index (conditional complexity)

    def create_or_retrieve_label(self):
        """

        :return: Label variable name and a map of fraud vs legit flag. e.g {"LEGIT" :[0], "FRAUD" :[1] }

        """
        labels = self.df["EVENT_LABEL"].astype(str).unique()
        true_labels = [str(l) for l in self.true_labels]
        false_labels = list(set(labels) - set(true_labels))

        for l in true_labels:
            self._fraud_detector_utils.create_or_update_label(str(l), "Fraud flag")

        for l in false_labels:
            self._fraud_detector_utils.create_or_update_label(str(l), "Legit flag")

        label_schema = {FRAUD_DETECTOR_LABEL_KEY_FRAUD: true_labels,
                        FRAUD_DETECTOR_LABEL_KEY_LEGIT: false_labels}

        return label_schema