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

sub-model test

Browse Source
This commit is contained in:
Mike J Innes 2017-02-01 18:23:28 +05:30
parent 4e27f2a405
commit e15a6e4896
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
test/basic.jl
View File

@ -8,3 +8,17 @@ let
@capture(syntax(d), _Frame(_Line(x_[1] * W_ + b_))) @capture(syntax(d), _Frame(_Line(x_[1] * W_ + b_)))
@test isa(x, Input) && isa(W, Param) && isa(b, Param) @test isa(x, Input) && isa(W, Param) && isa(b, Param)
end end
@net type TLP
first
second
function (x)
l1 = σ(first(x))
l2 = softmax(second(l1))
end
end
let a1 = Affine(10, 20), a2 = Affine(20, 15)
tlp = TLP(a1, a2)
@test tlp(xs) ≈ softmax(a2(σ(a1(xs))))
end