Update core/trainer.py

core/trainer.py

@@ -17,12 +17,12 @@ class GraphMambaTrainer:
         self.config = config
         self.device = device
         
-        # Balanced learning rate
+        # Optimal learning parameters
         self.lr = config['training']['learning_rate']
         self.epochs = config['training']['epochs']
-        self.patience = config['training'].get('patience', 12)
+        self.patience = config['training'].get('patience', 15)
         self.min_lr = config['training'].get('min_lr', 1e-6)
-        self.max_gap = config['training'].get('max_gap', 0.35)  # More lenient threshold
+        self.max_gap = config['training'].get('max_gap', 0.3)
         
         # Heavily regularized optimizer
         self.optimizer = optim.AdamW(