best val_rmse 0.3306

config.json

@@ -1,8 +1,9 @@
 {
-  "best_val_loss": 0.5102683305740356,
+  "best_val_loss": 0.33063554763793945,
   "comet_ai_key": null,
   "context_observations": {
     "add_rem": true,
+    "empirical_number_of_obs": 2,
     "max_num_obs": 15,
     "max_past": 5,
     "min_past": 3,
@@ -24,7 +25,7 @@
     "num_individuals": 10,
     "route_options": [
       "oral",
-      "oral"
+      "iv"
     ],
     "route_weights": [
       0.8,
@@ -33,7 +34,7 @@
     "same_route": true,
     "time": 0.0
   },
-  "experiment_dir": "/home/cesarali/Pharma/sim_priors_pk/results/comet/functional-flow-pk/d5c6cd390b1a42b4b4c757e82b18c16f",
+  "experiment_dir": "/home/cesarali/Pharma/sim_priors_pk/results/comet/functional-flow-pk/fd8d9b4637184613981f06969beb9e87",
   "experiment_indentifier": null,
   "experiment_name": "functional-flow-pk",
   "hf_model_card_path": [
@@ -94,7 +95,9 @@
       "simulated_ou_as_rates"
     ],
     "test_empirical_datasets": [
-      "cesarali/Indometacin"
+      "cesarali/lenuzza-2016",
+      "cesarali/Indometacin",
+      "cesarali/Theophylline"
     ],
     "test_protocol": "simulated",
     "test_size": 64,
@@ -154,11 +157,11 @@
   "run_index": 0,
   "tags": [
     "SIM_PRIOR",
-    "FlowPK",
-    "NodePK"
+    "FlowPK"
   ],
   "target_observations": {
     "add_rem": true,
+    "empirical_number_of_obs": 2,
     "max_num_obs": 15,
     "max_past": 5,
     "min_past": 3,
@@ -174,12 +177,12 @@
       0.9,
       0.999
     ],
-    "epochs": 20,
+    "epochs": 10,
     "eps": 1e-08,
     "gradient_clip_val": 1.0,
     "learning_rate": 0.0001,
-    "log_empirical_evaluation_pct": 0.25,
-    "log_image_every_epoch_pct": 0.25,
+    "log_empirical_evaluation_pct": 0.5,
+    "log_image_every_epoch_pct": 0.5,
     "log_interval": 1,
     "log_prediction_in_val": true,
     "log_reconstruction_in_val": true,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4a828d906b4c43cff0cd51009b2b1aab86551fa13f7eff8b09018576e39378a2
-size 39284635
+oid sha256:4bf6b15f609aa5046f88471b8f6ecbeec8a904e28a0378490b1aca592945ae6b
+size 39284311