TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

customisable loss

Browse Source
This commit is contained in:
Mike J Innes 2017-04-28 17:14:21 +01:00
parent 63b328142a
commit ef4ec5be4b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/utils.jl
View File

@ -7,7 +7,8 @@ initn(dims...) = randn(dims...)/100
tobatch(xs::Batch) = rawbatch(xs) tobatch(xs::Batch) = rawbatch(xs)
tobatch(xs) = tobatch(batchone(xs)) tobatch(xs) = tobatch(batchone(xs))
function train!(m, train, test = []; epoch = 1, η = 0.1) function train!(m, train, test = [];
epoch = 1, η = 0.1, loss = mse)
i = 0 i = 0
for e in 1:epoch for e in 1:epoch
info("Epoch $e") info("Epoch $e")
@ -16,7 +17,7 @@ function train!(m, train, test = []; epoch = 1, η = 0.1)
i += 1 i += 1
ŷ = m(x) ŷ = m(x)
any(isnan, ŷ) && error("NaN") any(isnan, ŷ) && error("NaN")
Δ = back!(mse, 1, ŷ, y) Δ = back!(loss, 1, ŷ, y)
back!(m, Δ, x) back!(m, Δ, x)
update!(m, η) update!(m, η)
i % 1000 == 0 && @show accuracy(m, test) i % 1000 == 0 && @show accuracy(m, test)