From 8c1e35f66f90de582e6792bfa11acc291137bf37 Mon Sep 17 00:00:00 2001
From: William Jeynes <william.jeynes@hotmail.com>
Date: Tue, 24 Mar 2026 13:16:18 +0000
Subject: [PATCH] Increase dropout on regression model to cut down on
 overfitting

---
 supporting/RAGAS_Service/train_regression.py | 6 ++----
 supporting/scorer/views/stats.py             | 2 +-
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/supporting/RAGAS_Service/train_regression.py b/supporting/RAGAS_Service/train_regression.py
index 2c5adc7..10c4bf7 100644
--- a/supporting/RAGAS_Service/train_regression.py
+++ b/supporting/RAGAS_Service/train_regression.py
@@ -14,8 +14,8 @@ import sys
 NUM_CLASSES    = 3
 EMBEDDING_MODEL = "all-mpnet-base-v2"
 HIDDEN_DIM     = 256
-DROPOUT        = 0.3
-LEARNING_RATE  = 1e-3
+DROPOUT        = 0.4
+LEARNING_RATE  = 2e-3
 WEIGHT_DECAY   = 1e-4
 BATCH_SIZE     = 64
 NUM_EPOCHS     = 30
@@ -90,8 +90,6 @@ class LogisticNet(nn.Module):
         return self.net(x)
 
 
-# ── Metrics ───────────────────────────────────────────────────────────────────
-
 def evaluate(model, loader, device):
     model.eval()
     all_preds, all_labels = [], []
diff --git a/supporting/scorer/views/stats.py b/supporting/scorer/views/stats.py
index eb41776..4c84052 100644
--- a/supporting/scorer/views/stats.py
+++ b/supporting/scorer/views/stats.py
@@ -64,7 +64,7 @@ def render():
         if ("flan" in file_path.name):
             thresh = 0.94
         if ("regression" in file_path.name):
-            thresh = 0.7
+            thresh = 0.75
 
         st.subheader(f"File: {file_path.name}")