Skip to content

Commit

Permalink
fix error msg in test
Browse files Browse the repository at this point in the history
  • Loading branch information
calad0i committed Apr 26, 2024
1 parent bb4f5ec commit a5f3a3f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/test_syn_hlayers.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit a5f3a3f

Please sign in to comment.