Upload config.json with huggingface_hub
Browse files- 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":
|
| 6 |
-
"
|
| 7 |
-
"
|
| 8 |
-
"lr_encoder": 2e-05,
|
| 9 |
-
"lr_heads": 0.001,
|
| 10 |
"weight_decay": 0.01,
|
| 11 |
-
"warmup_frac": 0.
|
| 12 |
"grad_clip": 1.0,
|
| 13 |
-
"
|
| 14 |
-
"
|
| 15 |
-
"
|
| 16 |
-
"
|
| 17 |
-
"
|
| 18 |
-
"
|
| 19 |
-
"
|
| 20 |
-
"
|
| 21 |
-
"
|
| 22 |
-
"fusion_dim":
|
| 23 |
-
"head_dim":
|
| 24 |
-
"
|
| 25 |
-
"best_spearman": 0.
|
| 26 |
-
"total_epochs_trained":
|
| 27 |
"class_weight": 239.96774193548387,
|
| 28 |
-
"
|
| 29 |
-
"
|
| 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 |
}
|