dheraingoud commited on
Commit
71f09be
·
verified ·
1 Parent(s): 94e6a61

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +20 -30
config.json CHANGED
@@ -1,36 +1,26 @@
1
  {
2
- "nt_model": "InstaDeepAI/nucleotide-transformer-v2-250m-multi-species",
3
- "max_token_len": 32,
4
  "num_features": 42,
5
- "batch_size": 32,
6
- "grad_accum": 2,
7
- "epochs": 12,
8
- "lr_encoder": 2e-05,
9
- "lr_heads": 0.001,
10
  "weight_decay": 0.01,
11
- "warmup_frac": 0.1,
12
  "grad_clip": 1.0,
13
- "freeze_epochs": 2,
14
- "unfreeze_layers": 4,
15
- "patience": 3,
16
- "loss_alpha": 0.45,
17
- "loss_beta": 0.2,
18
- "loss_gamma": 0.1,
19
- "loss_delta": 0.15,
20
- "loss_epsilon": 0.1,
21
- "dropout": 0.1,
22
- "fusion_dim": 512,
23
- "head_dim": 256,
24
- "cnn_channels": 64,
25
- "best_spearman": 0.47261583384770267,
26
- "total_epochs_trained": 12,
27
  "class_weight": 239.96774193548387,
28
- "hidden_size": 768,
29
- "version": "2.0",
30
- "fixes": [
31
- "robust_encoder_loading",
32
- "offtarget_label_masking",
33
- "rule_based_risk_regression",
34
- "expanded_features_42"
35
- ]
36
  }
 
1
  {
 
 
2
  "num_features": 42,
3
+ "batch_size": 128,
4
+ "epochs": 60,
5
+ "lr": 0.001,
 
 
6
  "weight_decay": 0.01,
7
+ "warmup_frac": 0.05,
8
  "grad_clip": 1.0,
9
+ "patience": 10,
10
+ "loss_alpha": 0.55,
11
+ "loss_beta": 0.1,
12
+ "loss_gamma": 0.15,
13
+ "loss_delta": 0.1,
14
+ "dropout": 0.15,
15
+ "cnn_channels": 128,
16
+ "feat_hidden": 256,
17
+ "feat_out": 128,
18
+ "fusion_dim": 384,
19
+ "head_dim": 192,
20
+ "version": "3.0",
21
+ "best_spearman": 0.6571680304412763,
22
+ "total_epochs_trained": 53,
23
  "class_weight": 239.96774193548387,
24
+ "architecture": "MultiScaleCNN_5branch_128ch + FeatureNet + DualHeads",
25
+ "total_params": 3062018
 
 
 
 
 
 
26
  }