Yak-hbdx commited on
Commit
cb828fa
·
verified ·
1 Parent(s): be40f3a

Push model using huggingface_hub.

Browse files
Files changed (1) hide show
  1. config.json +32 -1
config.json CHANGED
@@ -1,5 +1,27 @@
1
  {
2
  "main_config": {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  "model_config": {
4
  "class_mappings": [
5
  "12S_bin-1",
@@ -3888,6 +3910,12 @@
3888
  "vocab_size": 17,
3889
  "window": 2
3890
  },
 
 
 
 
 
 
3891
  "train_config": {
3892
  "batch_per_epoch": 99,
3893
  "batch_size": 512,
@@ -3910,6 +3938,9 @@
3910
  "shuffle_exp": false,
3911
  "top_k": 10,
3912
  "warmup_epoch": 10
3913
- }
 
 
 
3914
  }
3915
  }
 
1
  {
2
  "main_config": {
3
+ "cross_val": true,
4
+ "device_number": 1,
5
+ "inference": false,
6
+ "inference_settings": {
7
+ "infere_original_testset": false,
8
+ "model_path": "/nfs/home/yat_ldap/VS_Projects/TransfoRNA-Framework/models/tcga/TransfoRNA_FULL/sub_class/Seq/ckpt/model_params_tcga.pt",
9
+ "sequences_path": "/data/hbdx_ldap_local/analysis/data/sncRNA/test.h5ad"
10
+ },
11
+ "log_embedds": true,
12
+ "log_logits": false,
13
+ "model": {
14
+ "skorch_model": {
15
+ "_target_": "transforna.Net",
16
+ "batch_size": 512,
17
+ "criterion": "transforna.LossFunction",
18
+ "device": "cuda",
19
+ "iterator_train__shuffle": true,
20
+ "max_epochs": 3000,
21
+ "module": "transforna.GeneEmbeddModel",
22
+ "optimizer": "torch.optim.AdamW"
23
+ }
24
+ },
25
  "model_config": {
26
  "class_mappings": [
27
  "12S_bin-1",
 
3910
  "vocab_size": 17,
3911
  "window": 2
3912
  },
3913
+ "model_name": "seq",
3914
+ "num_replicates": 1,
3915
+ "path_to_id_models": "/nfs/home/yat_ldap/VS_Projects/TransfoRNA-Framework/models/tcga/",
3916
+ "seed": 0,
3917
+ "task": "tcga",
3918
+ "tensorboard": false,
3919
  "train_config": {
3920
  "batch_per_epoch": 99,
3921
  "batch_size": 512,
 
3938
  "shuffle_exp": false,
3939
  "top_k": 10,
3940
  "warmup_epoch": 10
3941
+ },
3942
+ "train_split": true,
3943
+ "trained_on": "full",
3944
+ "valid_size": 0.15
3945
  }
3946
  }