fix array mismatch

ourownstory · Jul 9, 2024 · e043201 · e043201
1 parent 5f103d8
commit e043201
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/neuralprophet/forecaster.py b/neuralprophet/forecaster.py
@@ -2916,7 +2916,13 @@ def _train(
 
         # Return metrics collected in logger as dataframe
         if self.metrics_logger.history is not None:
-            metrics_df = pd.DataFrame(self.metrics_logger.history)
+            # avoid array mismatch when continuing training
+            history = self.metrics_logger.history
+            max_length = max(len(lst) for lst in history.values())
+            for key in history:
+                while len(history[key]) < max_length:
+                    history[key].append(None)
+            metrics_df = pd.DataFrame(history)
         else:
             metrics_df = pd.DataFrame()
         return metrics_df