INNER CODE UNIT · Python
membership
AllAlgorithms/python · classification/fcm.py:17
def membership(self, data, centers):
U_temp = cdist(data, centers, 'euclidean')
U_temp = numpy.power(U_temp, 2/(self.m - 1))
denominator_ = U_temp.reshape(
(data.shape[0], 1, -1)).repeat(U_temp.shape[-1], axis=1)
denominator_ = U_temp[:, :, numpy.newaxis] / denominator_
return 1 / denominator_.sum(2)
def Centers(self, data, U):
um = U ** self.m
return (data.T @ um / numpy.sum(um, axis=0)).T
def newImage(self, U, centers, im):
best = numpy.argmax(self.U, axis=-1)
# print(best)
# numpy.round()
image = im.astype(int)
for i in range(256):