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 pandas as pd
import pyflux as pf
import scipy.stats as st
from pyFTS.common import SortedCollection, fts
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)
def train(self, data, **kwargs):
import pyflux as pf
if 'order' in kwargs:
order = kwargs.pop('order')
@ -95,8 +95,9 @@ class ARIMA(fts.FTS):
return ret
def forecast_distribution(self, data, **kwargs):
import pyflux as pf
sim_vector = self.inference(steps)
sim_vector = self.inference(1)
ret = []