strip down non-obvious exports

This commit is contained in:
Mike J Innes 2017-06-05 15:47:26 +01:00
parent eafba7a55f
commit 4685d2e672
6 changed files with 2 additions and 14 deletions

View File

@ -1,5 +1,3 @@
export onehot, onecold, chunk, partition, batches, sequences
"""
onehot('b', ['a', 'b', 'c', 'd']) => [false, true, false, false]
@ -27,8 +25,6 @@ onecold(y::AbstractMatrix, l...) =
using Iterators
import Iterators: Partition, partition
export partition
Base.length(l::Partition) = length(l.xs) ÷ l.step
_partition(r::UnitRange, step::Integer) = (step*(i-1)+1:step*i for i in 1:(r.stop÷step))

View File

@ -1,4 +1,4 @@
export σ, relu, softmax, flatten
export σ, relu, softmax
# Sigmoid
σ(x) = 1 ./ (1 + exp.(-x))

View File

@ -1,5 +1,3 @@
export mse, logloss
mse(, y) = sumabs2( .- y)/2
back!(::typeof(mse), Δ, , y) = Δ .* ( .- y)

View File

@ -1,4 +1,4 @@
export Conv2D, MaxPool, AvgPool, Reshape
export Conv2D
struct Conv2D <: Model
filter::Param{Array{Float64,4}} # [height, width, inchans, outchans]

View File

@ -1,5 +1,3 @@
export Model, back!, update!, param
# Basic model API
"""
@ -93,8 +91,6 @@ Base.copy!(p::Param, xs) = copy!(p.x, xs)
# Anonymous models
export Capacitor
struct Capacitor <: Model
graph::IVertex{Any}
end

View File

@ -1,5 +1,3 @@
export AArray, unsqueeze
call(f, xs...) = f(xs...)
# Arrays