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

trying logit crossentropy loss

Browse Source
This commit is contained in:
narawat lamaiin
2023-05-28 22:33:25 +07:00
parent 4fc4a5af91
commit fafaac9a45
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/learn.jl
View File

@@ -60,7 +60,7 @@ end
function compute_wRecChange!(n::lifNeuron, error::Float64) function compute_wRecChange!(n::lifNeuron, error::Float64)
n.eRec = n.phi * n.epsilonRec n.eRec = n.phi * n.epsilonRec
ΔwRecChange = n.eta * error * n.eRec ΔwRecChange = -n.eta * error * n.eRec
n.wRecChange .+= ΔwRecChange n.wRecChange .+= ΔwRecChange
reset_epsilonRec!(n) reset_epsilonRec!(n)
end end
@@ -69,7 +69,7 @@ function compute_wRecChange!(n::alifNeuron, error::Float64)
n.eRec_v = n.phi * n.epsilonRec n.eRec_v = n.phi * n.epsilonRec
n.eRec_a = -n.phi * n.beta * n.epsilonRecA n.eRec_a = -n.phi * n.beta * n.epsilonRecA
n.eRec = n.eRec_v + n.eRec_a n.eRec = n.eRec_v + n.eRec_a
ΔwRecChange = n.eta * error * n.eRec ΔwRecChange = -n.eta * error * n.eRec
n.wRecChange .+= ΔwRecChange n.wRecChange .+= ΔwRecChange
reset_epsilonRec!(n) reset_epsilonRec!(n)
reset_epsilonRecA!(n) reset_epsilonRecA!(n)
@@ -77,7 +77,7 @@ end
function compute_wRecChange!(n::linearNeuron, error::Float64) function compute_wRecChange!(n::linearNeuron, error::Float64)
n.eRec = n.phi * n.epsilonRec n.eRec = n.phi * n.epsilonRec
ΔwRecChange = n.eta * error * n.eRec ΔwRecChange = -n.eta * error * n.eRec
n.wRecChange .+= ΔwRecChange n.wRecChange .+= ΔwRecChange
reset_epsilonRec!(n) reset_epsilonRec!(n)
end end