config.py

N_FOLDS = 10

THRESHOLD = 0.5

VAL_SPLIT_SIZE = 0.2

TRAIN_DATA_DIR = '../data/train/'
TEST_DATA_DIR = '../data/test/'

TRAIN_MASK_DIR = '../data/train_mask'

CHECKPOINT_DIR = 'checkpoints/'