From 043c28d4805d322f4825bff54eca0f52d17a9b27 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Petr=C3=B4nio=20C=C3=A2ndido=20de=20Lima=20e=20Silva?= Date: Tue, 3 Jul 2018 18:57:00 -0300 Subject: [PATCH] Bugfix in cmsfts --- pyFTS/models/seasonal/cmsfts.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/pyFTS/models/seasonal/cmsfts.py b/pyFTS/models/seasonal/cmsfts.py index 45302cc..08d0e80 100644 --- a/pyFTS/models/seasonal/cmsfts.py +++ b/pyFTS/models/seasonal/cmsfts.py @@ -81,14 +81,18 @@ class ContextualMultiSeasonalFTS(sfts.SeasonalFTS): ndata = self.indexer.get_data(data) for k in np.arange(0, len(data)): + + if str(index[k]) in self.flrgs: - flrg = self.flrgs[str(index[k])] + flrg = self.flrgs[str(index[k])] - d = FuzzySet.get_fuzzysets(ndata[k], self.sets, ordered_sets, alpha_cut=self.alpha_cut) + d = FuzzySet.get_fuzzysets(ndata[k], self.sets, ordered_sets, alpha_cut=self.alpha_cut) - mp = self.get_midpoints(flrg, d) + mp = self.get_midpoints(flrg, d) - ret.append(sum(mp) / len(mp)) + ret.append(sum(mp) / len(mp)) + else: + ret.append(np.nan) return ret