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