Seq-TransfoRNA / conf /main_config.yaml
Yak-hbdx's picture
uploaded TransfoRNA repo
0b11a42 verified
raw
history blame
No virus
1.44 kB
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