From 05cecad3e40a4a619a486ff78cbb135c75d6cc7d Mon Sep 17 00:00:00 2001
From: RevathiJambunathan <revanathan@gmail.com>
Date: Tue, 16 Dec 2025 11:39:36 -0800
Subject: [PATCH 1/2] option to train on sim alone

---
 ml/Neural_Net_Classes.py | 4 +++-
 ml/train_model.py        | 2 ++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/ml/Neural_Net_Classes.py b/ml/Neural_Net_Classes.py
index f59a9702..c2c97642 100644
--- a/ml/Neural_Net_Classes.py
+++ b/ml/Neural_Net_Classes.py
@@ -136,6 +136,7 @@ def train_model(
         exp_inputs_val,
         exp_targets_val,
         num_epochs=1500,
+        train_on_exp=1
     ):
         for epoch in range(num_epochs):
             self.optimizer.zero_grad()
@@ -146,7 +147,8 @@ def train_model(
                 loss += nan_mse_loss(sim_targets, sim_outputs)
             if len(exp_inputs) > 0:
                 exp_outputs = self.calibrate(self(exp_inputs))
-                loss += nan_mse_loss(exp_targets, exp_outputs)
+                if train_on_exp==1:
+                    loss += nan_mse_loss(exp_targets, exp_outputs)
             loss.backward()
 
             self.optimizer.step()
diff --git a/ml/train_model.py b/ml/train_model.py
index 065b4343..c39feb3b 100644
--- a/ml/train_model.py
+++ b/ml/train_model.py
@@ -218,6 +218,7 @@
         model = CombinedNN(len(input_names), n_outputs, learning_rate=0.0001)
         model.to(device)  # moving to GPU
         NNmodel_start_time = time.time()
+        train_on_experiments = 1
         model.train_model(
             norm_sim_inputs_train.to(device),
             norm_sim_outputs_train.to(device),
@@ -228,6 +229,7 @@
             norm_expt_inputs_val.to(device),
             norm_expt_outputs_val.to(device),
             num_epochs=20000,
+            train_on_expt
         )
         NNmodel_end_time = time.time()
         print(f"Model_{i + 1} trained in ", NNmodel_end_time - NNmodel_start_time)

From fc6bebeb31f8c9668a529a7d77e3589241ed9728 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Tue, 16 Dec 2025 19:40:50 +0000
Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 ml/Neural_Net_Classes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ml/Neural_Net_Classes.py b/ml/Neural_Net_Classes.py
index c2c97642..1adbcf3e 100644
--- a/ml/Neural_Net_Classes.py
+++ b/ml/Neural_Net_Classes.py
@@ -136,7 +136,7 @@ def train_model(
         exp_inputs_val,
         exp_targets_val,
         num_epochs=1500,
-        train_on_exp=1
+        train_on_exp=1,
     ):
         for epoch in range(num_epochs):
             self.optimizer.zero_grad()
@@ -147,7 +147,7 @@ def train_model(
                 loss += nan_mse_loss(sim_targets, sim_outputs)
             if len(exp_inputs) > 0:
                 exp_outputs = self.calibrate(self(exp_inputs))
-                if train_on_exp==1:
+                if train_on_exp == 1:
                     loss += nan_mse_loss(exp_targets, exp_outputs)
             loss.backward()