speedup thanks to removing unnecessary np.atleast_2d

nicodv · Apr 28, 2015 · ac08188 · ac08188
1 parent 6950bd1
commit ac08188
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/kmodes/kmodes.py b/kmodes/kmodes.py
@@ -16,7 +16,7 @@ def get_max_value_key(dic):
 
 def matching_dissim(a, b):
     """Simple matching dissimilarity function"""
-    return (np.atleast_2d(a) != b).sum(axis=1)
+    return np.sum(a != b, axis=1)
 
 
 def init_huang(X, n_clusters):

diff --git a/kmodes/kprototypes.py b/kmodes/kprototypes.py
@@ -11,7 +11,7 @@
 
 def euclidean_dissim(a, b):
     """Euclidean distance dissimilarity function"""
-    return np.sum((np.atleast_2d(a) - b) ** 2, axis=1)
+    return np.sum((a - b) ** 2, axis=1)
 
 
 def move_point_num(point, ipoint, to_clust, from_clust,