We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 48c04de commit f979acbCopy full SHA for f979acb
tests/ignite/metrics/test_confusion_matrix.py
@@ -261,7 +261,7 @@ def test_miou():
261
262
res = iou_metric.compute().numpy()
263
264
- assert res == true_res_
+ assert pytest.approx(res) == true_res_
265
266
for ignore_index in range(3):
267
cm = ConfusionMatrix(num_classes=3)
@@ -271,7 +271,7 @@ def test_miou():
271
cm.update(output)
272
273
true_res_ = np.mean(true_res[:ignore_index] + true_res[ignore_index + 1 :])
274
- assert res == true_res_, f"{ignore_index}: {res} vs {true_res_}"
+ assert pytest.approx(res) == true_res_, f"{ignore_index}: {res} vs {true_res_}"
275
276
277
def test_cm_accuracy():
0 commit comments