MODEL_FORMATTER = {
    'base_model': 'Base model',
    'convolutional_hlr_1': 'Platt',
    'convolutional_hlr_5': 'Auxiliary',
    'fine_tune': 'Fine-tune',
    'MC_decoders': 'MC-Decoder',
    'MC_center': 'MC-Center'
}

LOSS_FORMATTER = {
    'CE': 'CrE',
    'CE-SD': 'CrE-SD',
    'SD': 'SD'
}

TABLE_FORMATTER = {
    'base_model': 'base model ',
    'convolutional_hlr_1': 'Platt      ',
    'convolutional_hlr_5': 'auxiliary  ',
    'fine_tune': 'fine-tune  ',
    'MC_decoders': 'MC-Decoder ',
    'MC_center': 'MC-Center  '
}

DATASET_FORMATTER = {
    "BRATS_2018": "BR18",
    "ISLES_2018": "IS18"
}
