def test_torch_autoencoder_simple_example(): mse = torch_autoencoder.simple_example() assert mse < 0.0001
def test_simple_example(): r2 = simple_example() assert r2 > 0.92