Skip to content

Commit

Permalink
docs: Updated the example for the second regression case to use a pow…
Browse files Browse the repository at this point in the history
…er of 2 in the loss.
  • Loading branch information
DoktorMike committed Jul 19, 2022
1 parent 450a76f commit ec2026b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/regression2.jl
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@ for epoch in 1:epochs
grads = Flux.gradient(pars) do
= m(x')
γ, ν, α, β = ŷ[1, :], ŷ[2, :], ŷ[3, :], ŷ[4, :]
trnloss = Statistics.mean(nigloss2(y, γ, ν, α, β, 0.01, 1))
trnloss = Statistics.mean(nigloss2(y, γ, ν, α, β, 0.01, 2))
trnloss
end
Flux.Optimise.update!(opt, pars, grads)
Expand Down

0 comments on commit ec2026b

Please sign in to comment.