Просмотр исходного кода

Add error message if make_future_dataframe before fit

Ben Letham 8 лет назад
Родитель
Сommit
7b9acd953f
2 измененных файлов с 5 добавлено и 0 удалено
  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') {
   if (freq == 'm') {
     freq <- 'month'
     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 <- seq(max(m$history.dates), length.out = periods + 1, by = freq)
   dates <- dates[2:(periods + 1)]  # Drop the first, which is max(history$ds)
   dates <- dates[2:(periods + 1)]  # Drop the first, which is max(history$ds)
   if (include_history) {
   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
         pd.Dataframe that extends forward from the end of self.history for the
         requested number of periods.
         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()
         last_date = self.history_dates.max()
         dates = pd.date_range(
         dates = pd.date_range(
             start=last_date,
             start=last_date,