File size: 1,062 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 | def classification_task(category, cv):
if category == 'Five_Class':
num_classes = 6
which_labels = 'Labels'
target_names = ['MASS_Malignant', 'MASS_Benign', 'MASS_BWC', 'CALC_Malignant', 'CALC_Benign', 'CALC_BWC']
excel_file_train = 'D:\\SSL_Breast_Lesion\\Labels\\Train_CV2.csv'
excel_file_validation = 'D:\\SSL_Breast_Lesion\\Labels\\Valid_CV2.csv'
excel_file_test = 'D:\\SSL_Breast_Lesion\\Labels\\Test_Full.csv'
return num_classes, target_names, excel_file_train, excel_file_validation, excel_file_test
def pretraining_task(category):
if category == 'Five_Class':
num_classes = 6
which_labels = 'Labels'
target_names = ['MASS_Malignant', 'MASS_Benign', 'MASS_BWC', 'CALC_Malignant', 'CALC_Benign', 'CALC_BWC']
excel_file_train = 'D:\\SSL_Breast_Lesion\\Labels\\Train_Full.csv'
excel_file_validation = 'na'
excel_file_test = 'na'
return num_classes, target_names, excel_file_train, excel_file_validation, excel_file_test
|