Restricting AdversarialDebiasing's trainable variables to current sco…

…pe (#255)
Trusted-AI · Sep 1, 2022 · efc6f9a · efc6f9a
1 parent 6023c26
commit efc6f9a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/aif360/algorithms/inprocessing/adversarial_debiasing.py b/aif360/algorithms/inprocessing/adversarial_debiasing.py
@@ -167,9 +167,9 @@ def fit(self, dataset):
             if self.debias:
                 adversary_opt = tf.train.AdamOptimizer(learning_rate)
 
-            classifier_vars = [var for var in tf.trainable_variables() if 'classifier_model' in var.name]
+            classifier_vars = [var for var in tf.trainable_variables(scope=self.scope_name) if 'classifier_model' in var.name]
             if self.debias:
-                adversary_vars = [var for var in tf.trainable_variables() if 'adversary_model' in var.name]
+                adversary_vars = [var for var in tf.trainable_variables(scope=self.scope_name) if 'adversary_model' in var.name]
                 # Update classifier parameters
                 adversary_grads = {var: grad for (grad, var) in adversary_opt.compute_gradients(pred_protected_attributes_loss,
                                                                                       var_list=classifier_vars)}