batching demo

This commit is contained in:
Mike J Innes 2016-10-29 00:12:53 +01:00
parent 8682052ffa
commit e807fc5c61
1 changed files with 26 additions and 0 deletions

26
examples/batching.jl Normal file
View File

@ -0,0 +1,26 @@
using Flux
# Traditional Approach
# 1000 samples of sequences of 100 28×28 3-colour images
rand(100, 10, 28, 28, 3)
# Basic Batching
data = Batch([collect(reshape(9(i-1):9i-1, 3, 3)) for i = 1:10])
Batch(flatten.(data))
data |> structure
Batch(flatten.(data)) |> structure
# Nested Batching
# DNA seqence, encoded as a list of [A, T, G, C]
x1 = Seq([[0,1,0,0], [1,0,0,0], [0,0,0,1]])
x2 = Seq([[0,0,1,0], [0,0,0,1], [0,0,1,0]])
data = Batch([x1, x2])
data |> structure