Update dissim.py

nicodv · Jun 10, 2019 · dca3d65 · dca3d65
1 parent fd89c66
commit dca3d65
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/kmodes/util/dissim.py b/kmodes/util/dissim.py
@@ -63,6 +63,10 @@ def jaccard_binary_dissim(a, b, **_):
 
     if np.isnan(a).any() or np.isnan(b).any():
         raise ValueError("Missing values detected in numerical columns.")
+    elif (a < 0).any() or (a > 1).any():
+        raise ValueError("Non-binary values detected in numerical columns.")
+    elif (b < 0).any() or (b > 1).any():
+        raise ValueError("Non-binary values detected in numerical columns.")
 
     i = np.sum((a == 1) & (b == 1), axis=1)
     j = np.sum((a == 1) | (b == 1), axis=1)