def __init__()

in preprocess/tweet_eval.py [0:0]

• 61 lines of code
• 8 McCabe index (conditional complexity)

    def __init__(self, subset_name):
        self.hf_identifier = "tweet_eval-" + subset_name
        self.subset_name = subset_name
        self.task_type = "classification"

        # for classification tasks, specify the meaning of each label
        if subset_name == "emoji":
            self.label = {
                0:"❤",
                1:"😍",
                2:"😂",
                3:"💕",
                4:"🔥",
                5:"😊",
                6:"😎",
                7:"✨",
                8:"💙",
                9:"😘",
                10:"📷",
                11:"🇺🇸",
                12:"☀",
                13:"💜",
                14:"😉",
                15:"💯",
                16:"😁",
                17:"🎄",
                18:"📸",
                19:"😜",
            }
        elif subset_name == "emotion":
            self.label = {
                0:"anger",
                1:"joy",
                2:"optimism",
                3:"sadness",
            }
        elif subset_name == "hate":
            self.label = {
                0:"non-hate",
                1:"hate",
            }
        elif subset_name == "irony":
            self.label = {
                0:"non-irony",
                1:"hate",
            }
        elif subset_name == "offensive":
            self.label = {
                0:"non-offensive",
                1:"hate",
            }
        elif subset_name == "sentiment":
            self.label = {
                0:"negative",
                1:"neutral",
                2:"positive",
            }
        else:
            self.label = {
                0:"none",
                1:"against",
                2:"favor",
            }