(variance: TensorType, lengthscales: TensorType)
| 92 | |
| 93 | @pytest.mark.parametrize("variance, lengthscales", [[2.3, 1.4]]) |
| 94 | def test_rbf_1d(variance: TensorType, lengthscales: TensorType) -> None: |
| 95 | X = rng.randn(3, 1) |
| 96 | kernel = SquaredExponential(lengthscales=lengthscales, variance=variance) |
| 97 | |
| 98 | gram_matrix = kernel(X) |
| 99 | reference_gram_matrix = ref_rbf_kernel(X, lengthscales, variance) |
| 100 | |
| 101 | assert_allclose(gram_matrix, reference_gram_matrix) |
| 102 | |
| 103 | |
| 104 | @pytest.mark.parametrize("variance, lengthscales", [[2.3, 1.4]]) |
nothing calls this directly
no test coverage detected
searching dependent graphs…