Remove vestigial comments, structure imports

This commit is contained in:
bl 2018-05-30 16:35:17 -07:00
parent 661ceedfa2
commit f8775433c9
6 changed files with 10 additions and 37 deletions

View file

@ -97,8 +97,6 @@ prophet <- function(df = NULL,
fit = TRUE,
...
) {
# fb-block 1
if (!is.null(changepoints)) {
n.changepoints <- length(changepoints)
}
@ -139,8 +137,6 @@ prophet <- function(df = NULL,
if ((fit) && (!is.null(df))) {
m <- fit.prophet(m, df, ...)
}
# fb-block 2
return(m)
}
@ -1509,5 +1505,3 @@ make_future_dataframe <- function(m, periods, freq = 'day',
}
return(data.frame(ds = dates))
}
# fb-block 3

View file

@ -17,7 +17,6 @@ import logging
import numpy as np
import pandas as pd
logger = logging.getLogger(__name__)

View file

@ -17,8 +17,12 @@ import warnings
import numpy as np
import pandas as pd
try:
import pystan # noqa F401
except ImportError:
logger.exception('You cannot run fbprophet without pystan installed')
# fb-block 1 start
from fbprophet.diagnostics import prophet_copy
from fbprophet.models import prophet_stan_model
from fbprophet.plot import (
plot,
@ -29,21 +33,11 @@ from fbprophet.plot import (
plot_yearly,
plot_seasonality,
)
from fbprophet.diagnostics import prophet_copy
# fb-block 1 end
logging.basicConfig()
logger = logging.getLogger(__name__)
warnings.filterwarnings("default", category=DeprecationWarning)
try:
import pystan # noqa F401
except ImportError:
logger.error('You cannot run prophet without pystan installed')
raise
# fb-block 2
class Prophet(object):
"""Prophet forecaster.
@ -903,7 +897,6 @@ class Prophet(object):
k = (L0 - L1) / T
return (k, m)
# fb-block 7
def fit(self, df, **kwargs):
"""Fit the Prophet model.
@ -1024,7 +1017,6 @@ class Prophet(object):
return self
# fb-block 8
def predict(self, df=None):
"""Predict using the prophet model.

View file

@ -11,22 +11,15 @@ from __future__ import print_function
from __future__ import unicode_literals
import pickle
# fb-block 1 start
import pkg_resources
# fb-block 1 end
# fb-block 2
def get_prophet_stan_model():
"""Load compiled Stan model"""
# fb-block 3
# fb-block 4 start
model_file = pkg_resources.resource_filename(
'fbprophet',
'stan_model/prophet_model.pkl',
)
# fb-block 4 end
with open(model_file, 'rb') as f:
return pickle.load(f)

View file

@ -11,13 +11,12 @@ from __future__ import print_function
from __future__ import unicode_literals
import itertools
import os
from unittest import TestCase
import numpy as np
import pandas as pd
# fb-block 1 start
import os
from unittest import TestCase
from fbprophet import Prophet
from fbprophet import diagnostics
@ -25,8 +24,6 @@ DATA_all = pd.read_csv(
os.path.join(os.path.dirname(__file__), 'data.csv'), parse_dates=['ds']
)
DATA = DATA_all.head(100)
# fb-block 1 end
# fb-block 2
class TestDiagnostics(TestCase):

View file

@ -10,12 +10,12 @@ from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
import os
from unittest import TestCase
import numpy as np
import pandas as pd
# fb-block 1 start
import os
from unittest import TestCase
from fbprophet import Prophet
DATA = pd.read_csv(
@ -26,8 +26,6 @@ DATA2 = pd.read_csv(
os.path.join(os.path.dirname(__file__), 'data2.csv'),
parse_dates=['ds'],
)
# fb-block 1 end
# fb-block 2
class TestProphet(TestCase):