add NaN filter

2023-07-23 08:32:31 +07:00
parent 3500313cba
commit a2badacfc5
1 changed files with 1 additions and 0 deletions
--- 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]