File size: 1,441 Bytes
0b11a42 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
defaults:
- model: transforna
- inference_settings: default
- override hydra/job_logging: disabled
task: tcga # tcga,sncrna or premirna or custom (for a custom dataset)
train_config:
_target_: train_model_configs.${task}.GeneEmbeddTrainConfig
model_config:
_target_: train_model_configs.${task}.GeneEmbeddModelConfig
#train settings
model_name: seq #seq, seq-seq, seq-struct, seq-reverse, or baseline
trained_on: full #full(production, more coverage) or id (for test/eval purposes)
path_to_models: /nfs/home/yat_ldap/VS_Projects/TransfoRNA-Framework/models/tcga/ #edit path to point to models/tcga/ directory: will be used if trained_on is full
inference: False # Should TransfoRNA be used for inference or train? True or False
#if inference is true, should the logits be logged?
log_logits: False
train_split: True # True or False
valid_size: 0.15 # 0 < valid_size < 1
#CV
cross_val: True # True or False
num_replicates: 1 # Integer, num_replicates for cross-validation
#seed
seed: 1 # Integer
device_number: 1 # Integer, select GPU
#logging sequence embeddings + metrics to tensorboard
log_embedds: True # True or False
tensorboard: False # True or False
#disable hydra output
hydra:
run:
dir: ./outputs/${now:%Y-%m-%d}/${model_name}
searchpath:
- file:///nfs/home/yat_ldap/VS_Projects/TransfoRNA-Framework/conf
# output_subdir: null #uncomment to disable hydra output
|