Procházet zdrojové kódy

Only allow a Prophet object to be fit once

Ben Letham před 8 roky
rodič
revize
34c0f80684
2 změnil soubory, kde provedl 6 přidání a 0 odebrání
  1. 3 0
      R/R/prophet.R
  2. 3 0
      python/fbprophet/forecaster.py

+ 3 - 0
R/R/prophet.R

@@ -473,6 +473,9 @@ logistic_growth_init <- function(df) {
 #'
 #' @export
 fit.prophet <- function(m, df, ...) {
+  if (!is.null(m$history)) {
+    stop("Prophet object can only be fit once. Instantiate a new object.")
+  }
   history <- df %>%
     dplyr::filter(!is.na(y))
   m$history.dates <- sort(zoo::as.Date(df$ds))

+ 3 - 0
python/fbprophet/forecaster.py

@@ -479,6 +479,9 @@ class Prophet(object):
         -------
         The fitted Prophet object.
         """
+        if self.history is not None:
+            raise Exception('Prophet object can only be fit once. '
+                            'Instantiate a new object.')
         history = df[df['y'].notnull()].copy()
         self.history_dates = pd.to_datetime(df['ds']).sort_values()