Skip to content

Commit 0329766

Browse files
committed
fix idkc
1 parent d5d4ba9 commit 0329766

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

ikpykit/cluster/_idkc.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -238,6 +238,7 @@ def _initialize_cluster(self, X):
238238
else:
239239
init_center = self.init_center
240240
for i in range(self.k):
241+
self.clusters_[i].set_center(init_center[i])
241242
self.clusters_[i].add_points(init_center[i], X[init_center[i]])
242243
self.data_index = np.delete(self.data_index, init_center)
243244
return self

ikpykit/cluster/_kcluster.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
"""
1010

1111
from collections.abc import Iterable
12+
from numbers import Integral
1213

1314
import numpy as np
1415
from scipy import sparse as sp
@@ -27,18 +28,14 @@ def set_center(self, center):
2728

2829
def add_points(self, ids, X):
2930
self.increment_kernel_mean_(X)
30-
if isinstance(ids, np.integer):
31-
if self.center is None:
32-
self.center = ids
31+
if isinstance(ids, Integral):
3332
self.points_.append(ids)
3433
elif isinstance(ids, Iterable):
35-
if self.center is None:
36-
raise ValueError("Cluster is not initialized.")
3734
self.points_.extend(ids)
3835

3936
def delete_points(self, points, X):
4037
self.reduce_kernel_mean_(X)
41-
if isinstance(points, np.integer):
38+
if isinstance(points, Integral):
4239
try:
4340
self.points_.remove(points)
4441
except:

0 commit comments

Comments
 (0)