File size: 1,507 Bytes
a639402 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
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 |