add avg updater

Saurav Agarwal · Saurav Agarwal · commit 375843b4175b · 2024-09-04T16:42:57.000-04:00
diff --git a/python/scripts/evaluators/constrained_learning.py b/python/scripts/evaluators/constrained_learning.py
@@ -109,6 +109,8 @@ def __init__(
             self.lambda_duals = np.array([1.0 / self.num_idfs for i in range(self.num_idfs)])
         elif dual_updater == "max_one":
             self.lambda_duals = np.array([1.0 / self.num_idfs for i in range(self.num_idfs)])
+        elif dual_updater == "avg":
+            self.lambda_duals = np.array([1.0 / self.num_idfs for i in range(self.num_idfs)])
         print(f"Initial Lambda_dual: {self.lambda_duals}")
 
         # Set the real values here
@@ -280,6 +282,10 @@ def fun_dual_updater(self, configs, lambdas):
 
             return lambdas
 
+        if dual_updater == "avg":
+            self.lambda_duals = np.array([1.0 / self.num_idfs for i in range(self.num_idfs)])
+            return lambdas
+
         raise ValueError("configs not recognized")
 
 
