rednaSander commited on
Commit
85a144d
·
verified ·
1 Parent(s): ea0df20

Upload proteomics/config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. proteomics/config.yaml +58 -0
proteomics/config.yaml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Training configuration for the OmicsFM checkpoint in this folder.
2
+ # Extracted verbatim from best_model.ckpt.
3
+
4
+ checkpoint: proteomics
5
+ training_run: prot_flashlfq_diann
6
+ epoch: 147
7
+ val_loss: 7.47722018494898
8
+ vocabulary_size: 20272
9
+ config:
10
+ data:
11
+ train_path: data\flashlfq_diann\train.h5ad
12
+ valid_path: data\flashlfq_diann\valid.h5ad
13
+ test_path: data\flashlfq_diann\test.h5ad
14
+ num_bins: 10
15
+ detect_groups: true
16
+ transfer:
17
+ checkpoint_path: null
18
+ freeze_transformer: false
19
+ freeze_epochs: 0
20
+ device: cuda
21
+ model:
22
+ d_model: 256
23
+ n_heads: 8
24
+ n_layers: 6
25
+ d_ff: 1024
26
+ dropout: 0.1
27
+ input_features: 1024
28
+ feature_context_ratio: 0.85
29
+ use_esmc: false
30
+ fasta_path: fasta/human_proteome_canonical_31032022.fasta
31
+ esmc_model: esmc_600m
32
+ esmc_cache: data/esmc_human_cache.pt
33
+ absent_species:
34
+ enabled: false
35
+ input_absent: 50
36
+ absent_context_ratio: 0.5
37
+ protein_groups:
38
+ enabled: false
39
+ max_group_size: 20
40
+ input_groups: 256
41
+ group_context_ratio: 0.2
42
+ encoder: mean
43
+ attention:
44
+ n_heads: 4
45
+ n_encoder_layers: 1
46
+ training:
47
+ batch_size: 64
48
+ learning_rate: 0.0001
49
+ weight_decay: 1.0e-05
50
+ max_epochs: 200
51
+ val_frequency: 1
52
+ early_stopping_patience: 5000
53
+ early_stopping_delta: 0.001
54
+ gradient_clip: 1.0
55
+ num_workers: 0
56
+ output:
57
+ save_dir: model
58
+ experiment_name: prot_flashlfq_diann