From a2badacfc542efb8978919f216aaba68a78bb06a Mon Sep 17 00:00:00 2001 From: ton Date: Sun, 23 Jul 2023 08:32:31 +0700 Subject: [PATCH] add NaN filter --- src/forward.jl | 1 + 1 file changed, 1 insertion(+) diff --git a/src/forward.jl b/src/forward.jl index 9aa504e..3682b35 100644 --- a/src/forward.jl +++ b/src/forward.jl @@ -91,6 +91,7 @@ function (kfn::kfn_1)(m::model, input_data::AbstractVector) end logit = [n.v_t1 for n in kfn.outputNeuronsArray] + logit = NaN ∉ logit ? normalize!(logit) : logit # _predict = Flux.softmax(logit) # predict = findall(isequal.(_predict, maximum(_predict)))[1]