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

Removing direct pyflux import from benchmark.BSTS modules

Browse Source
This commit is contained in:
Petrônio Cândido 2020-01-27 19:30:26 -03:00
parent 6a1ee719b7
commit 715772bbcc
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pyFTS/benchmarks/BSTS.py
View File

@ -3,7 +3,6 @@
import numpy as np import numpy as np
import pandas as pd import pandas as pd
import pyflux as pf
import scipy.stats as st import scipy.stats as st
from pyFTS.common import SortedCollection, fts from pyFTS.common import SortedCollection, fts
from pyFTS.probabilistic import ProbabilityDistribution from pyFTS.probabilistic import ProbabilityDistribution
@ -46,6 +45,7 @@ class ARIMA(fts.FTS):
self.shortname = "BSTS({},{},{})-{}".format(self.p,self.d,self.q,self.alpha) self.shortname = "BSTS({},{},{})-{}".format(self.p,self.d,self.q,self.alpha)
def train(self, data, **kwargs): def train(self, data, **kwargs):
import pyflux as pf
if 'order' in kwargs: if 'order' in kwargs:
order = kwargs.pop('order') order = kwargs.pop('order')
@ -95,8 +95,9 @@ class ARIMA(fts.FTS):
return ret return ret
def forecast_distribution(self, data, **kwargs): def forecast_distribution(self, data, **kwargs):
import pyflux as pf
sim_vector = self.inference(steps) sim_vector = self.inference(1)
ret = [] ret = []