This commit is contained in:
Petrônio Cândido 2018-11-05 14:32:19 -02:00
commit 96651d1059

View File

@ -42,6 +42,7 @@ class Retrainer(fts.FTS):
self.partitioner = self.partitioner_method(data=data, **self.partitioner_params) self.partitioner = self.partitioner_method(data=data, **self.partitioner_params)
self.model = self.fts_method(partitioner=self.partitioner, order=self.order, **self.fts_params) self.model = self.fts_method(partitioner=self.partitioner, order=self.order, **self.fts_params)
self.model.fit(data, **kwargs) self.model.fit(data, **kwargs)
self.shortname = self.model.shortname
def forecast(self, data, **kwargs): def forecast(self, data, **kwargs):
l = len(data) l = len(data)
@ -75,4 +76,4 @@ class Retrainer(fts.FTS):
:return: number of rules :return: number of rules
""" """
return len(self.model) return len(self.model)