From 283e9be5e0f97d457dbb06b31b4c314bada3905c Mon Sep 17 00:00:00 2001 From: Hasil Sharma Date: Thu, 22 Jun 2017 03:07:07 +0530 Subject: [PATCH] Set verbosity for Printing Meta Information (#197) * Set verbosity for Printing Meta Information (#188) * Fixed Indent and Removed Typos --- python/fbprophet/forecaster.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/python/fbprophet/forecaster.py b/python/fbprophet/forecaster.py index 8839a9d..ec11388 100644 --- a/python/fbprophet/forecaster.py +++ b/python/fbprophet/forecaster.py @@ -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