Bugfix in cmsfts

This commit is contained in:
Petrônio Cândido de Lima e Silva 2018-07-03 18:57:00 -03:00 committed by GitHub
parent 07bfceae78
commit 043c28d480
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -82,13 +82,17 @@ class ContextualMultiSeasonalFTS(sfts.SeasonalFTS):
for k in np.arange(0, len(data)):
flrg = self.flrgs[str(index[k])]
if str(index[k]) in self.flrgs:
d = FuzzySet.get_fuzzysets(ndata[k], self.sets, ordered_sets, alpha_cut=self.alpha_cut)
flrg = self.flrgs[str(index[k])]
mp = self.get_midpoints(flrg, d)
d = FuzzySet.get_fuzzysets(ndata[k], self.sets, ordered_sets, alpha_cut=self.alpha_cut)
ret.append(sum(mp) / len(mp))
mp = self.get_midpoints(flrg, d)
ret.append(sum(mp) / len(mp))
else:
ret.append(np.nan)
return ret