tqdm over eval dims

This commit is contained in:
Dr. Kashif Rasul
2020-01-15 16:59:31 +01:00
parent 7fd92875fe
commit 1987856470
+2 -1
View File
@@ -10,6 +10,7 @@ from typing import (
Union,
Callable,
)
from tqdm import tqdm
# Third-party imports
import numpy as np
@@ -544,7 +545,7 @@ class MultivariateEvaluator(Evaluator):
fcst_iterator, target_dimensionality + len(self.target_agg_funcs)
)
for dim in eval_dims:
for dim in tqdm(eval_dims):
agg_metrics, metrics_per_ts = super(MultivariateEvaluator, self).__call__(
self.extract_target_by_dim(ts_iterator_set[dim], dim),
self.extract_forecast_by_dim(fcst_iterator_set[dim], dim),