Przeglądaj źródła

Add error message if make_future_dataframe before fit

Ben Letham 7 lat temu
rodzic
commit
7b9acd953f
2 zmienionych plików z 5 dodań i 0 usunięć
  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,