diff --git a/pyFTS/models/incremental/Retrainer.py b/pyFTS/models/incremental/Retrainer.py index e8f9f54..50af370 100644 --- a/pyFTS/models/incremental/Retrainer.py +++ b/pyFTS/models/incremental/Retrainer.py @@ -42,6 +42,7 @@ class Retrainer(fts.FTS): 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) @@ -75,4 +76,4 @@ class Retrainer(fts.FTS): :return: number of rules """ - return len(self.model) \ No newline at end of file + return len(self.model)