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

@ -37,17 +37,11 @@ class Retrainer(fts.FTS):
self.is_high_order = True
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):
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.fit(data, **kwargs)
self.shortname = self.model.shortname
def forecast(self, data, **kwargs):
l = len(data)