def __init__(self, **kwargs):
self.centroids_distance = None
if 'load' in kwargs:
self._load(kwargs['load'])
else:
try:
method = kwargs['method']
del kwargs['method']
except KeyError as ke:
print(kr)
raise
if method == 'KMeans' or method == 'MiniBatchKMeans':
MiniBatchKMeans.__init__(self, **kwargs)
elif method == 'MeanShift':
MeanShift.__init__(self, **kwargs)
else:
e = "No method '{}' avaiable. Please use KMeans or MeanShift".format(
method)
log = Log(path='.', name='cluster_class')
log.write(error=e, data=self)
raise ValueError(e)
