From 358a6ea28790cd6eed60cb5f3ccaed71c3cdde41 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Petr=C3=B4nio=20C=C3=A2ndido?= Date: Tue, 11 Dec 2018 22:55:15 -0200 Subject: [PATCH] Improvements on cmvfts and pwfts for multivariate forecasting --- pyFTS/models/multivariate/cmvfts.py | 2 +- pyFTS/tests/multivariate.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/pyFTS/models/multivariate/cmvfts.py b/pyFTS/models/multivariate/cmvfts.py index ca31204..67baa86 100644 --- a/pyFTS/models/multivariate/cmvfts.py +++ b/pyFTS/models/multivariate/cmvfts.py @@ -86,7 +86,7 @@ class ClusteredMVFTS(mvfts.MVFTS): self.target_variable = var self.cluster.change_target_variable(var) self.model.partitioner = self.cluster - self.reset_calculated_values() + self.model.reset_calculated_values() ret[var.name] = self.model.forecast(ndata, fuzzyfied=self.pre_fuzzyfy, **kwargs) diff --git a/pyFTS/tests/multivariate.py b/pyFTS/tests/multivariate.py index f832ed8..f0f09e4 100644 --- a/pyFTS/tests/multivariate.py +++ b/pyFTS/tests/multivariate.py @@ -45,4 +45,6 @@ ax[0][1].scatter(forecasts['x'].values,forecasts['y'].values) ax[1][0].scatter(test['y'].values,test['x'].values) ax[1][0].scatter(forecasts['y'].values,forecasts['x'].values) ax[1][1].plot(test['y'].values) -ax[1][1].plot(forecasts['y'].values) \ No newline at end of file +ax[1][1].plot(forecasts['y'].values) + +print(forecasts) \ No newline at end of file