INNER CODE UNIT · Python

Centers

AllAlgorithms/python · classification/fcm.py:25

    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):
            image = numpy.where(image == float(i), centers[best[i]][0], image)
        return image

    def compute(self):
        self.U = self.membership(self.data, self.centers)

        past_U = numpy.copy(self.U)
        begin_time = datetime.datetime.now()

View source record →