diff --git a/examples/MNIST.jl b/examples/MNIST.jl index 88970f0f..e768e166 100644 --- a/examples/MNIST.jl +++ b/examples/MNIST.jl @@ -1,5 +1,5 @@ using Flux, MNIST -using Flux: accuracy +using Flux: accuracy, onehot data = [(trainfeatures(i), onehot(trainlabel(i), 0:9)) for i = 1:60_000] train = data[1:50_000]