|
|
|
|
|
|
| DATA_DIR = 'data'
|
|
|
| SCORE_FILE_NAME = 'scores.txt'
|
|
|
|
|
| MODEL_SAVE_BASE_PATH = 'trained_models_storage/model'
|
|
|
|
|
| DEFAULT_EPOCHS = 30
|
| DEFAULT_LR = 0.0001
|
| DEFAULT_BATCH_SIZE = 32
|
|
|
|
|
| DEFAULT_DROPOUT_RATE = 0.5
|
| DEFAULT_WEIGHT_DECAY = 1e-5
|
| DEFAULT_PCA_VARIANCE_RATIO = 0.99
|
|
|
|
|
| DEFAULT_OPTIMIZER = "Adam"
|
| DEFAULT_LR_SCHEDULER = "None"
|
| DEFAULT_SCHEDULER_PATIENCE = 5
|
| DEFAULT_SCHEDULER_FACTOR = 0.1
|
| DEFAULT_SCHEDULER_T_MAX = 50
|
|
|
|
|
| DEFAULT_LOSS_FUNCTION = "MSELoss"
|
|
|
|
|
| DEFAULT_EARLY_STOPPING_PATIENCE = 10
|
|
|
|
|
| VALIDATION_SPLIT_RATIO = 0.2
|
|
|
|
|
| DEFAULT_DATA_AUGMENTATION = True
|
|
|
