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

docstrings

Browse Source
This commit is contained in:
Mike J Innes 2018-01-17 16:12:12 +00:00
parent 8cca7accf2
commit bd57359535
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/data/mnist.jl
View File

@ -69,12 +69,33 @@ end
getfeatures(io::IO, index::Integer) = vec(getimage(io, index)) getfeatures(io::IO, index::Integer) = vec(getimage(io, index))
"""
images()
images(:test)
Load the MNIST images.
Each image is a 28×28 array of `Gray` colour values (see Colors.jl).
Returns the 60,000 training images by default; pass `:test` to retreive the
10,000 test images.
"""
function images(set = :train) function images(set = :train)
io = IOBuffer(read(set == :train ? TRAINIMAGES : TESTIMAGES)) io = IOBuffer(read(set == :train ? TRAINIMAGES : TESTIMAGES))
_, N, nrows, ncols = imageheader(io) _, N, nrows, ncols = imageheader(io)
[rawimage(io) for _ in 1:N] [rawimage(io) for _ in 1:N]
end end
"""
labels()
labels(:test)
Load the labels corresponding to each of the images returned from `images()`.
Each label is a number from 0-9.
Returns the 60,000 training labels by default; pass `:test` to retreive the
10,000 test labels.
"""
function labels(set = :train) function labels(set = :train)
io = IOBuffer(read(set == :train ? TRAINLABELS : TESTLABELS)) io = IOBuffer(read(set == :train ? TRAINLABELS : TESTLABELS))
_, N = labelheader(io) _, N = labelheader(io)