19 lines (16 with data), 556 Bytes

experiment {
  data_path = "../../data/conv_CTEPH302.pkl"
  output_dir = null # If it is None, output dir would be `output` folder under the data path directory
  batch_size = 16
  n_epochs = 100
  n_evals = 3
  n_bootstraps = 100
  n_folds = 6
  search_method = "grid search"
  model_name = "cox" # must be one of the function names in model_factory function of survival4D/models
}


# hyperparams search range define here.
# Notice that value name has to match input arguments' names, defined in survival4D/models
hypersearch {
  penalty_exp = [-2, 1]
}