iicd/naswot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update search.py

Browse Source
This commit is contained in:
Jack Turner
2020-06-03 15:23:26 +01:00
committed by GitHub
parent 63f4c2dbbb
commit 147e7f96f2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
search.py
View File

@@ -54,7 +54,7 @@ def get_batch_jacobian(net, x, target, to, device, args=None):
return jacob, target.detach() return jacob, target.detach()
def evidenceapprox_eval_score(jacob, labels=None): def eval_score(jacob, labels=None):
corrs = np.corrcoef(jacob) corrs = np.corrcoef(jacob)
v, _ = np.linalg.eig(corrs) v, _ = np.linalg.eig(corrs)
k = 1e-5 k = 1e-5
@@ -121,7 +121,7 @@ for N in runs:
jacobs = jacobs.reshape(jacobs.size(0), -1).cpu().numpy() jacobs = jacobs.reshape(jacobs.size(0), -1).cpu().numpy()
try: try:
s = evidenceapprox_eval_score(jacobs, labels) s = eval_score(jacobs, labels)
except Exception as e: except Exception as e:
print(e) print(e)
s = np.nan s = np.nan