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

run with what we have

Browse Source
This commit is contained in:
Mike J Innes 2016-08-25 17:26:52 +01:00
parent 29aab1e4e0
commit 4249e6e961
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
examples/MNIST.jl
View File

@ -1,13 +1,20 @@
using Flux, MNIST using Flux, MNIST, Flow, MacroTools
import Flux.MX: mxnet
import Flux: back!, update!, graph
const data = collect(zip([trainfeatures(i) for i = 1:60_000], @time begin
[onehot(trainlabel(i), 1:10) for i = 1:60_000])) const data = [(trainfeatures(i), onehot(trainlabel(i), 0:9)) for i = 1:60_000]
const train = data[1:50_000] const train = data[1:50_000]
const test = data[50_001:60_000] const test = data[50_001:60_000]
nothing
end
const m = Sequence( m = Chain(
Input(784), Input(784),
Dense(30), Sigmoid(), Dense(784, 128), relu,
Dense(10), Sigmoid()) Dense(128, 64), relu,
Dense(64, 10), softmax)
@time Flux.train!(m, train, test, epoch = 30) model = mxnet(m, 784)
@time Flux.train!(model, train, test, epoch = 1, η=0.001)