sam/pyFTS
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

FTS.predict bugfix for multivariate FTS

Browse Source
This commit is contained in:
Petrônio Cândido 2018-06-07 12:51:55 -03:00
parent 68a4a953b8
commit 409f9d5b6b
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pyFTS/tests/multivariate.py
View File

@ -7,7 +7,7 @@ from pyFTS.common import Transformations
from pyFTS.data import SONDA
df = SONDA.get_dataframe()
train = df.iloc[0:578241] #three years
#test = df.iloc[1572480:2096640] #ears
test = df.iloc[1572480:2096640] #one year
del df
from pyFTS.partitioners import Grid, Util as pUtil
@ -58,8 +58,10 @@ model1.target_variable = vavg
#model.fit(train, num_batches=60, save=True, batch_save=True, file_path='mvfts_sonda')
model1.fit(train, num_batches=200, save=True, batch_save=True, file_path='mvfts_sonda', distributed=True,
nodes=['192.168.0.110'], batch_save_interval=10)
#model1.fit(train, num_batches=200, save=True, batch_save=True, file_path='mvfts_sonda', distributed=False,
# nodes=['192.168.0.110'], batch_save_interval=10)
#model = Util.load_obj('mvfts_sonda')
model = Util.load_obj('mvfts_sonda')
forecasts = model.predict(test)