Commit 9e813861 authored by pawel rosikiewicz's avatar pawel rosikiewicz 💬
Browse files

Saved changes to: src/configs/config_functions.py src/configs/dataset_configs.py

parent a22040f8
Pipeline #187597 passed with stage
in 20 seconds
......@@ -130,8 +130,8 @@ def DEFINE_DATASETS(*,
"file_prefix": f'{module_name}',
"file_extension": f"encoded.npy",
"file_corename": {
"train": ["train_01", "train_02"],
"valid": "valid_01",
"train": "train",
"valid": "valid",
"test": "test_01",
"test_2": "test_02" # you may add more then one in a list !
}},
......
......@@ -72,7 +72,7 @@ CLASS_COLORS_zorder ={
# vs all other classes designated as NEGATIVE
CLASS_LABELS_CONFIGS = {
"Cancer_Detection_And_Classification":{
"cancer_detection_and_classification":{
"info":"more informative class names for raw data",
"class_labels_dict":{
......@@ -94,7 +94,7 @@ CLASS_LABELS_CONFIGS = {
'Melanoma': 'POSITIVE'}
},
"Cancer_Risk_Groups":{
"cancer_risk_groups":{
"info":"""
7 original classes were grouped into three oncological risk groups
with vasc&nv assigned into low lever skin lessions, all other cancer types into cancer benign,
......@@ -131,10 +131,17 @@ DATASET_CONFIGS = {
"HAM10000": {
"info": "raw data grouped with original classes, no augmentation, no duplicate removal",
"labels": [
"Cancer_Detection_And_Classification",
"Cancer_Risk_Groups"
"cancer_detection_and_classification",
"cancer_risk_groups"
]
}
},
"HAM10000_workshop": {
"info": "raw data grouped with original classes, no augmentation, no duplicate removal",
"labels": [
"cancer_detection_and_classification",
"cancer_risk_groups"
]
}
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment