diff --git a/pyFTS/models/hwang.py b/pyFTS/models/hwang.py index 6f862d7..dbfb050 100644 --- a/pyFTS/models/hwang.py +++ b/pyFTS/models/hwang.py @@ -19,6 +19,10 @@ class HighOrderFTS(fts.FTS): self.detail = "Hwang" def forecast(self, ndata, **kwargs): + + if 'order' in kwargs: + self.order = kwargs.get('order',self.order) + self.max_lag = self.order if self.sets == None: self.sets = self.partitioner.sets @@ -57,4 +61,4 @@ class HighOrderFTS(fts.FTS): return ret def train(self, data, **kwargs): - pass \ No newline at end of file + pass