Home

Welcome to the OutcomePrediction wiki!

• [MODEL]

• BaseModel --> Quick definition

• Model_Save_Path = "./"

• Batch_Size = 32

• RANDOM_SEED = 42

• loss_function = "CrossEntropyLoss"#"MSELoss"

• Max_Epochs = 2

• Precision = 32

• Backbone = "DenseNet121"#"UNet" #"densenet121"

• Pretrained = true

• Drop_Rate = 0.1

• wf = 4

• depth = 6

• activation = "Identity"

• inference = true

• emb_size = 1000

• [MODEL_PARAMETERS]

• spatial_dims = 3

• block_config = [1, 2, 4, 1]

• in_channels = 1

• out_channels = 1

• [OPTIMIZER]

• lr = 1e-4

• eps = 1e-7

• step_size = 5

• gamma = 0.1

• [SERVER]

• Project = "RTOG_test_upload"

• User = "yzhan"

• [DATA]

• DataFolder = "path/to/data"

• n_per_sample = 5000

• n_classes = 2

• n_channel = 3

• sub_patch_size = 16

• dim = 32, 256, 256

• vis = [0]

• train_size = 0.7

• val_size = 0.3

• target = "survival_months"

• Use_mask = false

• module = ["Anatomy"]

• [CRITERIA]

• arm = 2

• [MODALITY]

• Anatomy = "CT"

• Therapy = "Dose"

• [CHECKPOINT]

• monitor = "val_loss" #"val_acc_epoch"

• mode = "max"

• [AUGMENTATION]

• [VERBOSE]