example predictions

2016-10-04 21:11:03 +01:00 · 2016-10-04 21:11:03 +01:00 · 503265c964
parent 2609d47ce9
commit 503265c964
1 changed files with 6 additions and 0 deletions
--- a/examples/MNIST.jl
+++ b/examples/MNIST.jl
@ -13,4 +13,10 @@ m = Chain(
 # Convert to TensorFlow
 model = tf(m)

+# An example prediction pre-training
+model(data[1][1])
+
@time Flux.train!(model, train, test, η = 1e-3)
+
+# An example prediction post-training
+model(data[1][1])