Fix variable name (#341)

* Fix variable name

Signed-off-by: Daiki Katsuragawa <50144563+daikikatsuragawa@users.noreply.github.com>

* Fix variable name

Signed-off-by: Daiki Katsuragawa <50144563+daikikatsuragawa@users.noreply.github.com>

Signed-off-by: Daiki Katsuragawa <50144563+daikikatsuragawa@users.noreply.github.com>

Author: daikikatsuragawa

dice_ml/explainer_interfaces/feasible_base_vae.py

@@ -138,10 +138,10 @@ def train(self, pre_trained=False):
 
             train_dataset = torch.tensor(self.vae_train_feat).float()
             train_dataset = torch.utils.data.DataLoader(train_dataset, batch_size=self.batch_size, shuffle=True)
-            for train_x in enumerate(train_dataset):
+            for train in enumerate(train_dataset):
                 self.cf_vae_optimizer.zero_grad()
 
-                train_x = train_x[1]
+                train_x = train[1]
                 train_y = 1.0-torch.argmax(self.pred_model(train_x), dim=1)
                 train_size += train_x.shape[0]
 

dice_ml/explainer_interfaces/feasible_model_approx.py

@@ -83,10 +83,10 @@ def train(self, constraint_type, constraint_variables, constraint_direction, con
 
             train_dataset = torch.tensor(self.vae_train_feat).float()
             train_dataset = torch.utils.data.DataLoader(train_dataset, batch_size=self.batch_size, shuffle=True)
-            for train_x in enumerate(train_dataset):
+            for train in enumerate(train_dataset):
                 self.cf_vae_optimizer.zero_grad()
 
-                train_x = train_x[1]
+                train_x = train[1]
                 train_y = 1.0-torch.argmax(self.pred_model(train_x), dim=1)
                 train_size += train_x.shape[0]