sam/pyFTS
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

Update ensemble.py

Browse Source
This commit is contained in:
Petrônio Cândido de Lima e Silva 2019-02-27 14:16:18 -03:00 committed by GitHub
parent c559c1050d
commit c2c8a892f8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pyFTS/models/ensemble/ensemble.py
View File

@ -102,9 +102,8 @@ class EnsembleFTS(fts.FTS):
l = len(self.models) l = len(self.models)
if l == 1: if l == 1:
return forecasts[0] return forecasts[0]
w = [np.exp(-(l - k)) for k in range(l)] w = np.array([np.exp(-(l - k)) for k in range(l)])
sw = sum(w) w = w / np.nansum(w)
w = w / sw
ret = np.nansum([w[k] * forecasts[k] for k in range(l)]) ret = np.nansum([w[k] * forecasts[k] for k in range(l)])
return ret return ret