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)