From 6a6eb77d0df983966ae8b29a307e9992c4162a09 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Petr=C3=B4nio=20C=C3=A2ndido=20de=20Lima=20e=20Silva?= Date: Mon, 24 Oct 2022 15:19:37 -0300 Subject: [PATCH] Bugfix: Wrong number of partitions been generated by GridPartitioner --- pyFTS/partitioners/Grid.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pyFTS/partitioners/Grid.py b/pyFTS/partitioners/Grid.py index 2b1b14f..ee18bb6 100644 --- a/pyFTS/partitioners/Grid.py +++ b/pyFTS/partitioners/Grid.py @@ -22,11 +22,12 @@ class GridPartitioner(partitioner.Partitioner): kwargs = {'type': self.type, 'variable': self.variable} - dlen = self.max - self.min - partlen = dlen / self.partitions + #dlen = self.max - self.min + #partlen = dlen / self.partitions count = 0 - for c in np.arange(self.min, self.max, partlen): + centers, partlen = np.linspace(self.min, self.max, self.partitions, retstep=True) + for c in centers: _name = self.get_name(count) if self.membership_function == Membership.trimf: sets[_name] = FuzzySet.FuzzySet(_name, Membership.trimf, [c - partlen, c, c + partlen],c,**kwargs)