Skip to content

Commit 391bfc7

Browse files
authored
fixed bug in tf1 explainer (#327)
1 parent 7eeba1f commit 391bfc7

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

dice_ml/explainer_interfaces/dice_tensorflow1.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -346,7 +346,7 @@ def do_loss_initializations(self, yloss_type="hinge_loss", diversity_loss_type="
346346
feature_weights[feature] = round(1/normalized_mads[feature], 2)
347347

348348
feature_weights_list = []
349-
for feature in self.data_interface.encoded_feature_names:
349+
for feature in self.data_interface.ohe_encoded_feature_names:
350350
if feature in feature_weights:
351351
feature_weights_list.append(feature_weights[feature])
352352
else:

0 commit comments

Comments
 (0)