diff --git a/fastai_contrib/metrics.py b/fastai_contrib/metrics.py index e0f6fa5..4d1e22d 100644 --- a/fastai_contrib/metrics.py +++ b/fastai_contrib/metrics.py @@ -51,4 +51,8 @@ def accuracy_binary(input, targs): return (input == targs).float().mean() def mean_output(input, targs): + input = torch.nn.functional.softmax(input, dim=-1)[:, 1] return input.mean() + +def mean_target(input, targs): + return targs.float().mean()