constants.py

# For setup file
DEFAULT_SETUP_YAML_PATH = 'task_setup.yaml'
SETUP_YAML_LOCAL_DATA_DIR_KEY = 'data_dir'
SETUP_YAML_DOCKER_DATA_DIR_KEY = 'docker_data_dir'
SETUP_YAML_DIM_KEY = 'dim'
SETUP_YAML_TASKS_KEY = 'eval_tasks'
SETUP_YAML_EMB_KEY = 'emb_file'
SETUP_YAML_SPARK_MEM_KEY = 'spark_driver_memory'
SETUP_YAML_RESULTS_KEY = 'results_dir'

# Column names in train, test and embedding files
EMB_COL = 'embedding'
ID_COL = 'ImageID'
LABEL_COL = 'Confidence'

# For results
RESULT_FILE_PREFIX = 'result'

# General
RANDOM_SEED = 0