misc

2026-09-10 12:12:50 +08:00 · 2019-11-26 22:37:36 +08:00
parent 67da707648
commit 8d9f6cb60c
1 changed files with 4 additions and 0 deletions
@@ -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()