Merge pull request #153 from larroy/master

Default init for Kprototypes is documented as Cao
nicodv · Feb 12, 2021 · b276ca0 · b276ca0
2 parents 69689e2 + cbb64fd
commit b276ca0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/kmodes/kprototypes.py b/kmodes/kprototypes.py
@@ -421,7 +421,7 @@ class KPrototypes(kmodes.KModes):
     """
 
     def __init__(self, n_clusters=8, max_iter=100, num_dissim=euclidean_dissim,
-                 cat_dissim=matching_dissim, init='Huang', n_init=10, gamma=None,
+                 cat_dissim=matching_dissim, init='Cao', n_init=10, gamma=None,
                  verbose=0, random_state=None, n_jobs=1):
 
         super(KPrototypes, self).__init__(n_clusters, max_iter, cat_dissim, init,

diff --git a/setup.py b/setup.py
@@ -27,6 +27,9 @@
         'scipy>=0.13.3',
         'joblib>=0.11'
     ],
+    extras_requires=[
+        'nose',
+    ],
     classifiers=['Development Status :: 3 - Alpha',
                  'Intended Audience :: Science/Research',
                  'License :: OSI Approved :: MIT License',