src/utils/dataset_utils.py

def extract_target_names(method):

    if method in ['Calc_Mass_Malignant_MultiLabel']:
        target_names = ['Calcification', 'Mass', 'Malignant']

    elif method in ['CMMD_Malignant']:
        target_names = ['Benign', 'Malignant']

    elif method in ['MassRest']:
        target_names = ['Mass', 'No Mass'] 
    
    elif method in ['MassNormal']:
        target_names = ['Mass', 'No Finding', 'Others'] 

    elif method in ['No Defects']:
        target_names = ['No Defects', 'Defects']   

    elif method in ['Skinfolds', '2i', '3i']:
        target_names = ['No Skinfolds', 'Skinfolds']  

    elif method in ['Skinfold_MultiLabel']:
        target_names = ['Skinfolds', 'Defects']  

    elif method in ['Skinfold_Defect_MultiLabel']:
        target_names = ['2i Skinfolds', '3i Skinfolds', 'Defects'] 

    elif method in ['BreastDensity']:
        target_names = ['Density A', 'Density B', 'Density C', 'Density D'] 

    elif method in ['CM', 'CM_Malignant']:
        target_names = ['Normal', 'Benign', 'Malignant'] 

    elif method in ['CM_Malignant_Binary']:
        target_names = ['Non-Malignant', 'Malignant'] 

    elif method in ['Calc']:
        target_names = ['Normal', 'Benign', 'Malignant'] 

    elif method in ['Mass']:
        target_names = ['Normal', 'Benign', 'Malignant'] 

    elif method in ['MRI', 'MRI_MIL']:
        target_names = ['Normal', 'Benign', 'Malignant'] 
    
    else:
        raise ValueError("Method not correctly specified")
    return target_names