update pretrained model with 30 models

(5 cv folds , each with 6 repeats)
PMBio · May 16, 2024 · ddf752f · ddf752f
1 parent 7d92e9b
commit ddf752f
Show file tree

Hide file tree

Showing 41 changed files with 23 additions and 23 deletions.
diff --git a/pretrained_models/config.yaml b/pretrained_models/config.yaml
@@ -1,24 +1,24 @@
 model:
-    type: DeepSet
-    model_collection: agg_models
-    checkpoint: combined_agg.pt
-    config:
-        phi_layers: 2
-        phi_hidden_dim: 20
-        rho_layers: 3
-        rho_hidden_dim: 10
-        activation: LeakyReLU
-        pool: max
-        use_sigmoid: True
-        metrics:
-            objective: MSE
-            objective_mode: min
-            loss: MSE
-            all:
-                MSE: {}
-                PearsonCorrTorch: {}
-                MAE: {}
-                RSquared: {}
-        optimizer:
-            type: AdamW
-            config: {}
+  checkpoint: combined_agg.pt
+  config:
+    activation: LeakyReLU
+    metrics:
+      all:
+        MAE: {}
+        MSE: {}
+        PearsonCorrTorch: {}
+        RSquared: {}
+      loss: MSE
+      objective: MSE
+      objective_mode: min
+    optimizer:
+      config: {}
+      type: AdamW
+    phi_hidden_dim: 20
+    phi_layers: 2
+    pool: max
+    rho_hidden_dim: 10
+    rho_layers: 3
+    use_sigmoid: true
+  model_collection: agg_models
+  type: DeepSet
diff --git a/pretrained_models/models/repeat_0/best/bag_0.ckpt b/pretrained_models/models/repeat_0/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_1/best/bag_0.ckpt b/pretrained_models/models/repeat_1/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_1/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_1/best/bag_0.ckpt.reverse
diff --git a/pretrained_models/models/repeat_10/best/bag_0.ckpt b/pretrained_models/models/repeat_10/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_11/best/bag_0.ckpt b/pretrained_models/models/repeat_11/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_12/best/bag_0.ckpt b/pretrained_models/models/repeat_12/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_12/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_12/best/bag_0.ckpt.reverse
diff --git a/pretrained_models/models/repeat_13/best/bag_0.ckpt b/pretrained_models/models/repeat_13/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_14/best/bag_0.ckpt b/pretrained_models/models/repeat_14/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_15/best/bag_0.ckpt b/pretrained_models/models/repeat_15/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_16/best/bag_0.ckpt b/pretrained_models/models/repeat_16/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_16/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_16/best/bag_0.ckpt.reverse
diff --git a/pretrained_models/models/repeat_17/best/bag_0.ckpt b/pretrained_models/models/repeat_17/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_18/best/bag_0.ckpt b/pretrained_models/models/repeat_18/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_19/best/bag_0.ckpt b/pretrained_models/models/repeat_19/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_2/best/bag_0.ckpt b/pretrained_models/models/repeat_2/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_20/best/bag_0.ckpt b/pretrained_models/models/repeat_20/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_20/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_20/best/bag_0.ckpt.reverse
diff --git a/pretrained_models/models/repeat_21/best/bag_0.ckpt b/pretrained_models/models/repeat_21/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_22/best/bag_0.ckpt b/pretrained_models/models/repeat_22/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_23/best/bag_0.ckpt b/pretrained_models/models/repeat_23/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_23/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_23/best/bag_0.ckpt.reverse
diff --git a/pretrained_models/models/repeat_24/best/bag_0.ckpt b/pretrained_models/models/repeat_24/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_25/best/bag_0.ckpt b/pretrained_models/models/repeat_25/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_25/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_25/best/bag_0.ckpt.reverse
diff --git a/pretrained_models/models/repeat_26/best/bag_0.ckpt b/pretrained_models/models/repeat_26/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_27/best/bag_0.ckpt b/pretrained_models/models/repeat_27/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_27/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_27/best/bag_0.ckpt.reverse
diff --git a/pretrained_models/models/repeat_28/best/bag_0.ckpt b/pretrained_models/models/repeat_28/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_29/best/bag_0.ckpt b/pretrained_models/models/repeat_29/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_3/best/bag_0.ckpt b/pretrained_models/models/repeat_3/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_4/best/bag_0.ckpt b/pretrained_models/models/repeat_4/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_5/best/bag_0.ckpt b/pretrained_models/models/repeat_5/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_5/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_5/best/bag_0.ckpt.reverse
diff --git a/pretrained_models/models/repeat_6/best/bag_0.ckpt b/pretrained_models/models/repeat_6/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_6/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_6/best/bag_0.ckpt.reverse
diff --git a/pretrained_models/models/repeat_7/best/bag_0.ckpt b/pretrained_models/models/repeat_7/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_7/best/bag_0.ckpt.reverse b/pretrained_models/models/repeat_7/best/bag_0.ckpt.reverse
diff --git a/pretrained_models/models/repeat_8/best/bag_0.ckpt b/pretrained_models/models/repeat_8/best/bag_0.ckpt
diff --git a/pretrained_models/models/repeat_9/best/bag_0.ckpt b/pretrained_models/models/repeat_9/best/bag_0.ckpt