TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Tests added.

Browse Source
This commit is contained in:
Manjunath Bhat 2019-03-12 02:36:37 +05:30 committed by GitHub
parent 633f0df01f
commit 61386c04f8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
test/layers/stateless.jl
View File

@ -49,7 +49,31 @@ const ϵ = 1e-7
@testset "logitbinarycrossentropy" begin @testset "logitbinarycrossentropy" begin
@test logitbinarycrossentropy.(logŷ, y) ≈ binarycrossentropy.(σ.(logŷ), y; ϵ=0) @test logitbinarycrossentropy.(logŷ, y) ≈ binarycrossentropy.(σ.(logŷ), y; ϵ=0)
end end
y = [1 2 3]
y1 = [4.0 5.0 6.0]
@testset "KLDivergence" begin
@test Flux.KLDivergence(y, y1) ≈ 4.761838062403337
@test Flux.KLDivergence(y, y) ≈ 0
end
@testset "Hinge" begin
@test Flux.Hinge(y, y1) ≈ 0
@test Flux.Hinge(y, 0.2 .* y) ≈ 0.33333
end
y = [0.1 0.2 0.3]
y1 = [0.4 0.5 0.6]
@testset "Poisson" begin
@test Flux.Poisson(y, y1) ≈ 1.0160455586700767
@test Flux.Poisson(y, y) ≈ 0.5044459776946685
end
@testset "logcosh" begin
@test Flux.logcosh(y, y1) ≈ 0.13302230977782092
@test Flux.logcosh(y, y) ≈ 0
end
@testset "no spurious promotions" begin @testset "no spurious promotions" begin
for T in (Float16, Float32, Float64) for T in (Float16, Float32, Float64)
y = rand(T, 2) y = rand(T, 2)