Update IncrementalEnsemble.py

This commit is contained in:
Petrônio Cândido de Lima e Silva 2019-02-27 14:44:58 -03:00 committed by GitHub
parent 3e88c94823
commit d05ed762fe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -72,7 +72,7 @@ class IncrementalEnsembleFTS(ensemble.EnsembleFTS):
if k >= self.window_length: if k >= self.window_length:
data_window.pop(0) data_window.pop(0)
if k % self.batch_size == 0 and k >= self.window_length: if k % self.batch_size == 0 and k - self.max_lag >= self.window_length:
self.train(data_window, **kwargs) self.train(data_window, **kwargs)
sample = data[k - self.max_lag: k] sample = data[k - self.max_lag: k]