From a5f3a3f6f0adf65ac2fc975c5a6e389df6f20536 Mon Sep 17 00:00:00 2001 From: Chang Sun Date: Fri, 26 Apr 2024 01:28:11 -0700 Subject: [PATCH] fix error msg in test --- test/test_syn_hlayers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_syn_hlayers.py b/test/test_syn_hlayers.py index 19ac529..52db42a 100644 --- a/test/test_syn_hlayers.py +++ b/test/test_syn_hlayers.py @@ -57,7 +57,7 @@ def get_data(N: int, sigma: float, max_scale: float, seed): def softmax_cond(proxy, hls): match_precent = np.mean(np.argmax(proxy, axis=1) == np.argmax(hls, axis=1)) - assert match_precent > 0.90, f"Keras-Proxy mismatch: {(1-match_precent) * 100}% of samples are different. Sample: {proxy[:5]} vs {hls[:5]}" + assert match_precent > 0.90, f"Proxy-HLS4ML mismatch: {(1-match_precent) * 100}% of samples are different. Sample: {proxy[:5]} vs {hls[:5]}" def custom_fn(x):