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

example predictions

Browse Source
This commit is contained in:
Mike J Innes 2016-10-04 21:11:03 +01:00
parent 2609d47ce9
commit 503265c964
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/MNIST.jl
View File

@ -13,4 +13,10 @@ m = Chain(
# Convert to TensorFlow # Convert to TensorFlow
model = tf(m) model = tf(m)
# An example prediction pre-training
model(data[1][1])
@time Flux.train!(model, train, test, η = 1e-3) @time Flux.train!(model, train, test, η = 1e-3)
# An example prediction post-training
model(data[1][1])