Bugfix on Retrainer.shortname

This commit is contained in:
Petrônio Cândido de Lima e Silva 2018-11-01 17:54:53 -03:00 committed by GitHub
parent 307836f261
commit b2a889a73e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -36,18 +36,12 @@ class Retrainer(fts.FTS):
"""The batch interval between each retraining""" """The batch interval between each retraining"""
self.is_high_order = True self.is_high_order = True
self.uod_clip = False self.uod_clip = False
@property
def shortname(self):
if self.model is None:
self.model = self.fts_method()
return self.model.shortname
def train(self, data, **kwargs): def train(self, data, **kwargs):
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)