From be93972d7a2ae7e8604b1b1982e2026ba2b28623 Mon Sep 17 00:00:00 2001 From: Lorenzo Stella Date: Fri, 24 Jan 2025 23:23:47 +0100 Subject: [PATCH 1/3] fix JET complaint --- src/losses.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/losses.jl b/src/losses.jl index 54f63a4..bdcf21c 100644 --- a/src/losses.jl +++ b/src/losses.jl @@ -9,7 +9,7 @@ function cross_entropy(probs, label; dims = 1, agg = mean) return agg(.-sum(ce_summands; dims = dims)) end -function class_error(probs, label; dims = 1, agg = mean) +function class_error(probs::AbstractArray, label::AbstractArray; dims = 1, agg = mean) class_predicted = argmax(probs; dims = dims) class_actual = argmax(label; dims = dims) return agg(1 .- (class_predicted .== class_actual)) From 443213c455e5a8a31f3d0c81e9d483a639247a96 Mon Sep 17 00:00:00 2001 From: Lorenzo Stella Date: Fri, 24 Jan 2025 23:33:08 +0100 Subject: [PATCH 2/3] fix --- src/losses.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/losses.jl b/src/losses.jl index bdcf21c..6bcad30 100644 --- a/src/losses.jl +++ b/src/losses.jl @@ -9,7 +9,7 @@ function cross_entropy(probs, label; dims = 1, agg = mean) return agg(.-sum(ce_summands; dims = dims)) end -function class_error(probs::AbstractArray, label::AbstractArray; dims = 1, agg = mean) +function class_error(probs::Array, label::Array; dims = 1, agg = mean) class_predicted = argmax(probs; dims = dims) class_actual = argmax(label; dims = dims) return agg(1 .- (class_predicted .== class_actual)) From b8e1872e1ed170aef7fcf0aee04b94e29846934f Mon Sep 17 00:00:00 2001 From: Lorenzo Stella Date: Fri, 24 Jan 2025 23:41:10 +0100 Subject: [PATCH 3/3] Update Project.toml --- Project.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Project.toml b/Project.toml index 494a4b0..90666cf 100644 --- a/Project.toml +++ b/Project.toml @@ -21,4 +21,4 @@ LinearAlgebra = "1.9" NNlib = "0.9" Statistics = "1.9" Zygote = "0.6" -julia = "1.9" +julia = "1.10"