diff --git a/neon/transforms/cost.py b/neon/transforms/cost.py index cb714020..9e9125b6 100644 --- a/neon/transforms/cost.py +++ b/neon/transforms/cost.py @@ -438,7 +438,7 @@ def __init__(self, num_classes, binarize=False, epsilon=1e-6): self.bin_buf = None self.eps = epsilon - def __call__(self, y, t): + def __call__(self, y, t, calcrange=slice(0, None)): """ Returns a numpy array with the precision and recall metrics.