16 lines
332 B
Julia
16 lines
332 B
Julia
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),
|
|
Dense(30),
|
|
Sigmoid(),
|
|
Dense(10),
|
|
Sigmoid())
|
|
|
|
Flux.train!(m, epoch = 30)
|