saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-07-05 04:17:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Replace assertions with error messages

Browse source
This commit is contained in:
bl 2018-05-30 14:36:36 -07:00
parent 1b91fa3064
commit b96800e423
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
python/fbprophet/forecaster.py
View file

@ -267,7 +267,11 @@ class Prophet(object):
else:
df['floor'] = 0
if self.growth == 'logistic':
assert 'cap' in df
if 'cap' not in df:
raise ValueError(
"Capacities must be supplied for logistic growth in "
"column 'cap'"
)
df['cap_scaled'] = (df['cap'] - df['floor']) / self.y_scale
df['t'] = (df['ds'] - self.start) / self.t_scale
@ -522,7 +526,8 @@ class Prophet(object):
prior_scale = float(self.holidays_prior_scale)
if mode is None:
mode = self.seasonality_mode
assert prior_scale > 0
if prior_scale <= 0:
raise ValueError('Prior scale must be > 0')
if mode not in ['additive', 'multiplicative']:
raise ValueError("mode must be 'additive' or 'multiplicative'")
self.extra_regressors[name] = {