Bladeren bron

Add error message if make_future_dataframe before fit

Ben Letham 8 jaren geleden
bovenliggende
commit
7b9acd953f
2 gewijzigde bestanden met toevoegingen van 5 en 0 verwijderingen
  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,