From 5565250c28d87eefd694e42ff67f68c2ffec8a35 Mon Sep 17 00:00:00 2001 From: Adarsh Kumar <45385384+AdarshKumar712@users.noreply.github.com> Date: Mon, 2 Mar 2020 13:46:33 +0530 Subject: [PATCH] Updated test for tversky --- test/layers/stateless.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/layers/stateless.jl b/test/layers/stateless.jl index c09d1aae..702288b6 100644 --- a/test/layers/stateless.jl +++ b/test/layers/stateless.jl @@ -99,7 +99,7 @@ const ϵ = 1e-7 @testset "tversky_loss" begin @test Flux.tversky_loss(y,y1) ≈ 0.028747433264887046 - @test Flux.tversky_loss(y,y1,0.8) ≈ 0.050200803212851364 + @test Flux.tversky_loss(y,y1,beta = 0.8) ≈ 0.050200803212851364 @test Flux.tversky_loss(y,y) ≈ -0.5576923076923075 end