From c24ebe6b81f8ef9f37a35834faf06e4674ec3969 Mon Sep 17 00:00:00 2001 From: sam Date: Tue, 6 Aug 2024 15:18:31 +0400 Subject: [PATCH] remove np.array in cMeans --- pyFTS/partitioners/CMeans.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyFTS/partitioners/CMeans.py b/pyFTS/partitioners/CMeans.py index df67219..b447ade 100644 --- a/pyFTS/partitioners/CMeans.py +++ b/pyFTS/partitioners/CMeans.py @@ -87,11 +87,11 @@ class CMeansPartitioner(partitioner.Partitioner): kwargs = {'type': self.type, 'variable': self.variable} centroides = c_means(self.partitions, data, 1) + centroides = [v[0] for v in centroides] centroides.append(self.max) centroides.append(self.min) - centroides = list(set(centroides)) centroides.sort() - for c in np.arange(1, len(centroides) - 1): + for c in range(1, len(centroides) - 1): _name = self.get_name(c) sets[_name] = FuzzySet.FuzzySet(_name, Membership.trimf, [round(centroides[c - 1], 3), round(centroides[c], 3), round(centroides[c + 1], 3)],