mirror of
https://github.com/saymrwulf/prophet.git
synced 2026-09-16 22:20:23 +00:00
Remove vestigial comments, structure imports
This commit is contained in:
parent
661ceedfa2
commit
f8775433c9
6 changed files with 10 additions and 37 deletions
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -17,7 +17,6 @@ import logging
|
|||
import numpy as np
|
||||
import pandas as pd
|
||||
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -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.
|
||||
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
|
||||
|
|
|
|||
|
|
@ -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):
|
||||
|
|
|
|||
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Reference in a new issue