From 147e7f96f2716948f09c3726d3cfabb1df3adbd4 Mon Sep 17 00:00:00 2001 From: Jack Turner Date: Wed, 3 Jun 2020 15:23:26 +0100 Subject: [PATCH] Update search.py --- search.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/search.py b/search.py index 2d65c69..fc40248 100644 --- a/search.py +++ b/search.py @@ -54,7 +54,7 @@ def get_batch_jacobian(net, x, target, to, device, args=None): return jacob, target.detach() -def evidenceapprox_eval_score(jacob, labels=None): +def eval_score(jacob, labels=None): corrs = np.corrcoef(jacob) v, _ = np.linalg.eig(corrs) k = 1e-5 @@ -121,7 +121,7 @@ for N in runs: jacobs = jacobs.reshape(jacobs.size(0), -1).cpu().numpy() try: - s = evidenceapprox_eval_score(jacobs, labels) + s = eval_score(jacobs, labels) except Exception as e: print(e) s = np.nan