From 8d9f6cb60cba21913e52deb77513e551b4d6fb97 Mon Sep 17 00:00:00 2001 From: wassname Date: Tue, 26 Nov 2019 22:37:36 +0800 Subject: [PATCH] misc --- fastai_contrib/metrics.py | 4 ++++ 1 file changed, 4 insertions(+) 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()