INNER CODE UNIT · Python

best

AllAlgorithms/python · classification/fcm.py:30

        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()
        for i in range(self.max_iter):

            self.centers = self.Centers(self.data, self.U)
            self.U = self.membership(self.data, self.centers)

View source record →