File tree Expand file tree Collapse file tree 2 files changed +4
-6
lines changed Expand file tree Collapse file tree 2 files changed +4
-6
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change 99"""
1010
1111from collections .abc import Iterable
12+ from numbers import Integral
1213
1314import numpy as np
1415from 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 :
You can’t perform that action at this time.
0 commit comments