1 parent 5f103d8 commit e043201Copy full SHA for e043201
1 file changed
neuralprophet/forecaster.py
@@ -2916,7 +2916,13 @@ def _train(
2916
2917
# Return metrics collected in logger as dataframe
2918
if self.metrics_logger.history is not None:
2919
- metrics_df = pd.DataFrame(self.metrics_logger.history)
+ # avoid array mismatch when continuing training
2920
+ history = self.metrics_logger.history
2921
+ max_length = max(len(lst) for lst in history.values())
2922
+ for key in history:
2923
+ while len(history[key]) < max_length:
2924
+ history[key].append(None)
2925
+ metrics_df = pd.DataFrame(history)
2926
else:
2927
metrics_df = pd.DataFrame()
2928
return metrics_df
0 commit comments