in occant_utils/metrics.py [0:0]
def reduce_metrics(metrics, reduction):
if type(metrics) == type({}):
r_metrics = {}
if reduction == "mean":
for k, v in metrics.items():
r_metrics[k] = v.mean()
elif reduction == "sum":
for k, v in metrics.items():
r_metrics[k] = v.sum()
else:
for k, v in metrics.items():
r_metrics[k] = v
else:
if reduction == "mean":
r_metrics = metrics.mean()
elif reduction == "sum":
r_metrics = metrics.sum()
else:
r_metrics = metrics
return r_metrics