2016-05-10 16:06:31 +00:00
|
|
|
using Flux, MNIST
|
|
|
|
|
|
|
|
|
|
const data = collect(zip([trainfeatures(i) for i = 1:60_000],
|
|
|
|
|
[onehot(trainlabel(i), 1:10) for i = 1:60_000]))
|
|
|
|
|
const train = data[1:50_000]
|
|
|
|
|
const test = data[50_001:60_000]
|
|
|
|
|
|
|
|
|
|
const m = Sequence(
|
|
|
|
|
Input(784),
|
2016-05-10 18:01:38 +00:00
|
|
|
Dense(30), Sigmoid(),
|
|
|
|
|
Dense(10), Sigmoid())
|
2016-05-10 16:06:31 +00:00
|
|
|
|
2016-05-10 18:01:38 +00:00
|
|
|
@time Flux.train!(m, train, test, epoch = 30)
|
