treex.metrics.Reduction

An enumeration.

Source code in treex/metrics/reduce.py

class Reduction(enum.Enum):
    sum = enum.auto()
    sum_over_batch_size = enum.auto()
    weighted_mean = enum.auto()