|
18 | 18 | @test repr(k) == "Mahalanobis Kernel (size(P) = $(size(P)))"
|
19 | 19 |
|
20 | 20 | M1, M2 = rand(rng,3,2), rand(rng,3,2)
|
21 |
| - fdm = FiniteDifferences.Central(5, 1); |
22 |
| - |
23 | 21 |
|
24 | 22 | function FiniteDifferences.to_vec(dist::SqMahalanobis{Float64})
|
25 | 23 | return vec(dist.qmat), x -> SqMahalanobis(reshape(x, size(dist.qmat)...))
|
|
30 | 28 | return MahalanobisKernel(P=Array(U'*U))(v1, v2)
|
31 | 29 | end
|
32 | 30 |
|
33 |
| - @test all(FiniteDifferences.j′vp(fdm, test_mahakernel, a, U, v1, v2)[1] .≈ |
| 31 | + @test all(FiniteDifferences.j′vp(FDM, test_mahakernel, a, U, v1, v2)[1] .≈ |
34 | 32 | UpperTriangular(Zygote.pullback(test_mahakernel, U, v1, v2)[2](a)[1]))
|
35 | 33 |
|
36 | 34 | function test_sqmaha(U::UpperTriangular, v1::AbstractVector, v2::AbstractVector)
|
37 | 35 | return SqMahalanobis(Array(U'*U))(v1, v2)
|
38 | 36 | end
|
39 | 37 |
|
40 |
| - @test all(FiniteDifferences.j′vp(fdm, test_sqmaha, a, U, v1, v2)[1] .≈ |
| 38 | + @test all(FiniteDifferences.j′vp(FDM, test_sqmaha, a, U, v1, v2)[1] .≈ |
41 | 39 | UpperTriangular(Zygote.pullback(test_sqmaha, U, v1, v2)[2](a)[1]))
|
42 | 40 |
|
43 | 41 | # test_ADs(U -> MahalanobisKernel(P=Array(U' * U)), U, ADs=[:Zygote])
|
|
0 commit comments