diff --git a/src/Batches/iter.jl b/src/Batches/iter.jl index 27fd6c26..885c9318 100644 --- a/src/Batches/iter.jl +++ b/src/Batches/iter.jl @@ -4,7 +4,10 @@ using Base.Iterators: partition partitionr(xs, n) = take(partition(xs, n), length(xs)÷n) +chunk(xs, n) = (partitionr(xs, length(xs)÷n)...,) + batches(xs, n) = (Batch([xs...]) for xs in partitionr(xs, n)) +seqs(xs, n) = (Seq([xs...]) for xs in partitionr(xs, n)) # Stateful iteration diff --git a/src/data.jl b/src/data.jl index bc6369c1..a50be7a4 100644 --- a/src/data.jl +++ b/src/data.jl @@ -21,5 +21,3 @@ onecold(y::AbstractVector, labels = 1:length(y)) = onecold(y::AbstractMatrix, l...) = squeeze(mapslices(y -> onecold(y, l...), y, 2), 2) - -chunk(xs, n) = Base.Iterators.partition(xs, length(xs)÷n)