ton/Ironpen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add NaN filter

Browse Source
This commit is contained in:
ton
2023-07-23 08:32:31 +07:00
parent 3500313cba
commit a2badacfc5
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/forward.jl
View File

@@ -91,6 +91,7 @@ function (kfn::kfn_1)(m::model, input_data::AbstractVector)
end end
logit = [n.v_t1 for n in kfn.outputNeuronsArray] logit = [n.v_t1 for n in kfn.outputNeuronsArray]
logit = NaN ∉ logit ? normalize!(logit) : logit
# _predict = Flux.softmax(logit) # _predict = Flux.softmax(logit)
# predict = findall(isequal.(_predict, maximum(_predict)))[1] # predict = findall(isequal.(_predict, maximum(_predict)))[1]