Explorar o código

Set verbosity for Printing Meta Information (#197)

* Set verbosity for Printing Meta Information (#188)

* Fixed Indent and Removed Typos
Hasil Sharma %!s(int64=8) %!d(string=hai) anos
pai
achega
283e9be5e0
Modificáronse 1 ficheiros con 10 adicións e 5 borrados
  1. 10 5
      python/fbprophet/forecaster.py

+ 10 - 5
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