We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3803f7e commit 2106faeCopy full SHA for 2106fae
1 file changed
test/test_syn_hlayers.py
@@ -56,6 +56,8 @@ def get_data(N: int, sigma: float, max_scale: float, seed):
56
57
58
def softmax_cond(proxy, hls):
59
+ if len(proxy) < 100:
60
+ return True
61
match_precent = np.mean(np.argmax(proxy, axis=1) == np.argmax(hls, axis=1))
62
assert match_precent > 0.90, f"Proxy-HLS4ML mismatch: {(1-match_precent) * 100}% of samples are different. Sample: {proxy[:5]} vs {hls[:5]}"
63
0 commit comments