new axis

backend/disentangle_concepts.py

@@ -104,7 +104,8 @@ def regenerate_images(model, z, decision_boundary, min_epsilon=-3, max_epsilon=3
     return images, lambdas
 
 def generate_joint_effect(model, z, decision_boundaries, min_epsilon=-3, max_epsilon=3, count=5, latent_space='Z'):
-    decision_boundary_joint = np.sum(decision_boundaries, axis=1)
+    decision_boundary_joint = np.sum(decision_boundaries, axis=0)
+    print(decision_boundary_joint.shape)
     return regenerate_images(model, z, decision_boundary_joint, min_epsilon=min_epsilon, max_epsilon=max_epsilon, count=count, latent_space=latent_space)
     
 def generate_original_image(z, model, latent_space='Z'):