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

add tests

This commit is contained in:
cossio 2020-05-05 19:05:04 +02:00
parent 480473a81b
commit 06c1e20372
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/layers/stateless.jl
View File

@ -44,6 +44,7 @@ end
lossvalue = 1.203972804325936 lossvalue = 1.203972804325936
@testset "crossentropy" begin @testset "crossentropy" begin
@test crossentropy([0.1,0.0,0.9], [0.1,0.0,0.9]) ≈ crossentropy([0.1,0.9], [0.1,0.9])
@test crossentropy(ŷ, y) ≈ lossvalue @test crossentropy(ŷ, y) ≈ lossvalue
end end
@ -76,6 +77,7 @@ end
y = [1 2 3] y = [1 2 3]
ŷ = [4.0 5.0 6.0] ŷ = [4.0 5.0 6.0]
@testset "kldivergence" begin @testset "kldivergence" begin
@test Flux.kldivergence([0.1,0.0,0.9], [0.1,0.0,0.9]) ≈ Flux.kldivergence([0.1,0.9], [0.1,0.9])
@test Flux.kldivergence(ŷ, y) ≈ -1.7661057888493457 @test Flux.kldivergence(ŷ, y) ≈ -1.7661057888493457
@test Flux.kldivergence(y, y) ≈ 0 @test Flux.kldivergence(y, y) ≈ 0
end end