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):
238
238
else :
239
239
init_center = self .init_center
240
240
for i in range (self .k ):
241
+ self .clusters_ [i ].set_center (init_center [i ])
241
242
self .clusters_ [i ].add_points (init_center [i ], X [init_center [i ]])
242
243
self .data_index = np .delete (self .data_index , init_center )
243
244
return self
Original file line number Diff line number Diff line change 9
9
"""
10
10
11
11
from collections .abc import Iterable
12
+ from numbers import Integral
12
13
13
14
import numpy as np
14
15
from scipy import sparse as sp
@@ -27,18 +28,14 @@ def set_center(self, center):
27
28
28
29
def add_points (self , ids , X ):
29
30
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 ):
33
32
self .points_ .append (ids )
34
33
elif isinstance (ids , Iterable ):
35
- if self .center is None :
36
- raise ValueError ("Cluster is not initialized." )
37
34
self .points_ .extend (ids )
38
35
39
36
def delete_points (self , points , X ):
40
37
self .reduce_kernel_mean_ (X )
41
- if isinstance (points , np . integer ):
38
+ if isinstance (points , Integral ):
42
39
try :
43
40
self .points_ .remove (points )
44
41
except :
You can’t perform that action at this time.
0 commit comments