diff --git a/tests/eva/vision/metrics/defaults/segmentation/test_multiclass.py b/tests/eva/vision/metrics/defaults/segmentation/test_multiclass.py index a86ec9ec..64b848c8 100644 --- a/tests/eva/vision/metrics/defaults/segmentation/test_multiclass.py +++ b/tests/eva/vision/metrics/defaults/segmentation/test_multiclass.py @@ -13,7 +13,9 @@ PREDS_ONE = torch.randint(0, NUM_CLASSES_ONE, (NUM_BATCHES, BATCH_SIZE, 32, 32)) TARGET_ONE = torch.randint(0, NUM_CLASSES_ONE, (NUM_BATCHES, BATCH_SIZE, 32, 32)) EXPECTED_ONE = { - "GeneralizedDiceScore": torch.tensor(0.3482658863067627), + "DiceScore (micro)": torch.tensor(0.3482658863067627), + "DiceScore (macro)": torch.tensor(0.34805023670196533), + "DiceScore (weighted)": torch.tensor(0.3484232723712921), "MeanIoU": torch.tensor(0.2109210342168808), } """Test features."""