modAL.density.information_density bug fixed

cosmic-cortex · cosmic-cortex · commit 65dcf3cb345c · 2018-02-10T10:50:26.000+01:00
diff --git a/modAL/density.py b/modAL/density.py
@@ -54,7 +54,7 @@ def information_density(X, similarity_measure=cosine_similarity):
 
     """
     inf_density = np.zeros(shape=(len(X),))
-    for X_idx, X in enumerate(X):
-        inf_density[X_idx] = sum(similarity_measure(X, X_j) for X_j in X)
+    for X_idx, X_inst in enumerate(X):
+        inf_density[X_idx] = sum(similarity_measure(X_inst, X_j) for X_j in X)
 
     return inf_density/len(X)
