add benchmark for ng_dissim

nicodv · Nov 15, 2017 · 5fdbea3 · 5fdbea3
1 parent 7eea6aa
commit 5fdbea3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/examples/benchmark_kmodes.py b/examples/benchmark_kmodes.py
@@ -1,8 +1,11 @@
 #!/usr/bin/env python
 
 import timeit
+
 import numpy as np
+
 from kmodes.kmodes import KModes
+from kmodes.util.dissim import ng_dissim
 
 # number of clusters
 K = 20
@@ -20,13 +23,17 @@ def huang():
     KModes(n_clusters=K, init='Huang', n_init=1, verbose=2).fit_predict(data)
 
 
+def huang_ng_dissim():
+    KModes(n_clusters=K, init='Huang', cat_dissim=ng_dissim, n_init=1, verbose=2).fit_predict(data)
+
+
 def cao():
     KModes(n_clusters=K, init='Cao', verbose=2).fit_predict(data)
 
 
 if __name__ == '__main__':
 
-    for cm in ('huang', 'cao'):
+    for cm in ('huang', 'huang_ng_dissim', 'cao'):
         print(cm.capitalize() + ': {:.2} seconds'.format(
             timeit.timeit(cm + '()',
                           setup='from __main__ import ' + cm,

diff --git a/examples/benchmark_kprototypes.py b/examples/benchmark_kprototypes.py
@@ -1,7 +1,9 @@
 #!/usr/bin/env python
 
 import timeit
+
 import numpy as np
+
 from kmodes.kprototypes import KPrototypes
 
 # number of clusters