saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-07-24 19:43:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Set verbosity for Printing Meta Information (#197)

Browse source 
* Set verbosity for Printing Meta Information (#188)

* Fixed Indent and Removed Typos
This commit is contained in:
Hasil Sharma 2017-06-22 03:07:07 +05:30 committed by Ben Letham
parent 1d5cb784c4
commit 283e9be5e0
1 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
python/fbprophet/forecaster.py
View file

@ -12,6 +12,9 @@ from __future__ import unicode_literals
from collections import defaultdict
from datetime import timedelta
import logging
logger = logging.getLogger(__name__)
from matplotlib import pyplot as plt
from matplotlib.dates import MonthLocator, num2date
@ -27,11 +30,13 @@ from fbprophet.models import prophet_stan_models
try:
import pystan
except ImportError:
print('You cannot run prophet without pystan installed')
logger.error('You cannot run prophet without pystan installed')
raise
# fb-block 2
class Prophet(object):
"""Prophet forecaster.
@ -377,8 +382,8 @@ class Prophet(object):
if self.yearly_seasonality == 'auto':
if last - first < pd.Timedelta(days=730):
self.yearly_seasonality = False
print('Disabling yearly seasonality. Run prophet with '
'yearly_seasonality=True to override this.')
logger.info('Disabling yearly seasonality. Run prophet with '
'yearly_seasonality=True to override this.')
else:
self.yearly_seasonality = True
if self.weekly_seasonality == 'auto':
@ -387,8 +392,8 @@ class Prophet(object):
if ((last - first < pd.Timedelta(weeks=2)) or
(min_dt >= pd.Timedelta(weeks=1))):
self.weekly_seasonality = False
print('Disabling weekly seasonality. Run prophet with '
'weekly_seasonality=True to override this.')
logger.info('Disabling weekly seasonality. Run prophet with '
'weekly_seasonality=True to override this.')
else:
self.weekly_seasonality = True