diff --git a/kmodes/kprototypes.py b/kmodes/kprototypes.py index aa037f6..e30e8e8 100644 --- a/kmodes/kprototypes.py +++ b/kmodes/kprototypes.py @@ -499,13 +499,13 @@ def _k_prototypes_iter(Xnum, Xcat, centroids, cl_attr_sum, cl_memb_sum, cl_attr_ ) cl_attr_freq, membship, centroids[1] = kmodes._move_point_cat( Xcat[rindx], rindx, old_clust, from_clust, - cl_attr_freq, membship, centroids[1] + cl_attr_freq, membship, centroids[1], weight ) return centroids, cl_attr_sum, cl_memb_sum, cl_attr_freq, membship, moves -def _move_point_num(point, to_clust, from_clust, cl_attr_sum, cl_memb_sum, sample_weight=1): +def _move_point_num(point, to_clust, from_clust, cl_attr_sum, cl_memb_sum, sample_weight): """Move point between clusters, numerical attributes.""" # Update sum of attributes in cluster. for iattr, curattr in enumerate(point):