Merge pull request Trusted-AI#84 from IBM/fix_lfr_distances_shapes

fix lfr helper with correct dists shape.
gkumbhat · Jun 6, 2019 · cd418e3 · cd418e3
2 parents 7010561 + 006cf2a
commit cd418e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aif360/algorithms/preprocessing/lfr_helpers/helpers.py b/aif360/algorithms/preprocessing/lfr_helpers/helpers.py
@@ -4,7 +4,7 @@
 
 @jit
 def distances(X, v, alpha, N, P, k):
-    dists = np.zeros((N, P))
+    dists = np.zeros((N, k))
     for i in range(N):
         for p in range(P):
             for j in range(k):