From 2c9ca4d133cc04389f637fecfade946184bc35fa Mon Sep 17 00:00:00 2001 From: Junhao Liang <43094337+darkliang@users.noreply.github.com> Date: Sat, 23 Oct 2021 07:54:50 +0800 Subject: [PATCH] Fix wrong index --- lib/utils/evaluation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/utils/evaluation.py b/lib/utils/evaluation.py index 84e531a..fba16af 100644 --- a/lib/utils/evaluation.py +++ b/lib/utils/evaluation.py @@ -94,7 +94,7 @@ def identification(data, dataset, metric='euc'): de_diag(acc[1, :, :, i]), de_diag(acc[2, :, :, i]))) result_dict["scalar/test_accuracy/NM"] = acc[0, :, :, i] - result_dict["scalar/test_accuracy/BG"] = acc[0, :, :, i] + result_dict["scalar/test_accuracy/BG"] = acc[1, :, :, i] result_dict["scalar/test_accuracy/CL"] = acc[2, :, :, i] np.set_printoptions(precision=2, floatmode='fixed') for i in range(1):