Browse Source

Add error message if make_future_dataframe before fit

Ben Letham 7 years ago
parent
commit
7b9acd953f
2 changed files with 5 additions and 0 deletions
  1. 3 0
      R/R/prophet.R
  2. 2 0
      python/fbprophet/forecaster.py

+ 3 - 0
R/R/prophet.R

@@ -1379,6 +1379,9 @@ make_future_dataframe <- function(m, periods, freq = 'day',
   if (freq == 'm') {
     freq <- 'month'
   }
+  if (is.null(m$history.dates)) {
+    stop('Model must be fit before this can be used.')
+  }
   dates <- seq(max(m$history.dates), length.out = periods + 1, by = freq)
   dates <- dates[2:(periods + 1)]  # Drop the first, which is max(history$ds)
   if (include_history) {

+ 2 - 0
python/fbprophet/forecaster.py

@@ -1214,6 +1214,8 @@ class Prophet(object):
         pd.Dataframe that extends forward from the end of self.history for the
         requested number of periods.
         """
+        if self.history_dates is None:
+            raise Exception('Model must be fit before this can be used.')
         last_date = self.history_dates.max()
         dates = pd.date_range(
             start=last_date,