def get_num_classes()

in svhn_based_exp/utils.py [0:0]

• 10 lines of code
• 5 McCabe index (conditional complexity)

def get_num_classes(task):
    if task in ["uu-ent-down", "dd-mis-down", "uu-ent-up1", "uu-ent-up2", "dd-mis-up"]:
        num_classes = 2
    elif task == "lf-mis-up":
        num_classes = 1
    elif task == "lf-mis-down":
        num_classes = 4
    elif task in ["hs-mis-up", "hs-mis-down"]:
        num_classes = 10
    return num_classes