return work code

2024-08-08 11:59:33 +04:00 · 2024-08-08 11:59:33 +04:00 · 8be936e383
commit 8be936e383
parent e1f72797ea
1 changed files with 1 additions and 1 deletions
--- a/pyFTS/partitioners/KMeans.py
+++ b/pyFTS/partitioners/KMeans.py
@ -87,9 +87,9 @@ class KMeansPartitioner(partitioner.Partitioner):
        kwargs = {'type': self.type, 'variable': self.variable}

        centroides = k_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 range(1, len(centroides) - 1):
            _name = self.get_name(c)