def __create_labels()

in frauddetector/profiler.py [0:0]

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

    def __create_labels(self, data, event_column):
        """Create target labels for AFD
            
            Args:
                data (pandas.core.frame.DataFrame): panda's dataframe to create summary statistics for
                event_column (str): column that contains the target event
            Returns:
                label_list (list): List of dicts with label names
        """
        if len(data[event_column].unique()) > 2:
            logging.error(f"Target column {event_column} has more than 2 unique values! Please review your data and fix the {event_column} content!")
        labels = data[event_column].unique().tolist()
        label_list = [{"name": x} for x in labels]
        return label_list