upd

YieldsLabs · Sep 20, 2024 · 0c2a82f · 0c2a82f
1 parent 961a2ca
commit 0c2a82f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/infrastructure/event_dispatcher/load_balancer.py b/infrastructure/event_dispatcher/load_balancer.py
@@ -57,6 +57,9 @@ def determine_priority_group(self, priority: int) -> int:
 
         control_outputs = self._pid.update(errors)
 
-        return np.random.choice(
-            np.arange(len(control_outputs)), p=softmax(control_outputs)
-        )
+        penalty_factor = np.exp(-processed_ratios)
+        weighted_outputs = control_outputs * penalty_factor
+
+        smooth_outputs = softmax(weighted_outputs)
+
+        return np.random.choice(np.arange(len(smooth_outputs)), p=smooth_outputs)