Setting stan backend to None: ~95% memory usage decrease

carlomazzaferro · Mar 27, 2020 · 12ccbbe · 12ccbbe
1 parent 5b29cf8
commit 12ccbbe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hts/model/p.py b/hts/model/p.py
@@ -19,7 +19,6 @@
 
 
 class FBProphetModel(TimeSeriesModel):
-    # TODO: include history to calculate residuals
     def __init__(self, node: HierarchyTree, **kwargs):
         super().__init__(Model.prophet.name, node, **kwargs)
         self.cap = None
@@ -59,6 +58,7 @@ def fit(self, **fit_args) -> 'TimeSeriesModel':
         df = self._reformat(self.node.item)
         with suppress_stdout_stderr():
             self.model = self.model.fit(df)
+            self.model.stan_backend = None
         return self
 
     def predict(self,