def reduce_metrics()

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