def test_zero_div(): mae = MeanAbsoluteError() with pytest.raises( NotComputableError, match= r"MeanAbsoluteError must have at least one example before it can be computed" ): mae.compute()
def test_compute(): mae = MeanAbsoluteError() y_pred = torch.Tensor([[2.0], [-2.0]]) y = torch.zeros(2) mae.update((y_pred, y)) assert mae.compute() == 2.0 mae.reset() y_pred = torch.Tensor([[3.0], [-3.0]]) y = torch.zeros(2) mae.update((y_pred, y)) assert mae.compute() == 3.0
def test_zero_div(): mae = MeanAbsoluteError() with pytest.raises(NotComputableError): mae.compute()