diff options
| author | Tom Smeding <tom@tomsmeding.com> | 2024-10-26 21:31:00 +0200 |
|---|---|---|
| committer | Tom Smeding <tom@tomsmeding.com> | 2024-10-26 21:31:00 +0200 |
| commit | 8cbc20cb1759a7a8c0e11f7da91da79cedf3d2d6 (patch) | |
| tree | 16865a8a58ccb96257114cbd6ee1f83cadff94dc /test | |
| parent | 633302b54e90d4b34f4a717327167c196171a250 (diff) | |
Can differentiate 'neural' now
Diffstat (limited to 'test')
| -rw-r--r-- | test/Main.hs | 26 |
1 files changed, 13 insertions, 13 deletions
diff --git a/test/Main.hs b/test/Main.hs index 11d15b4..a3a614a 100644 --- a/test/Main.hs +++ b/test/Main.hs @@ -270,19 +270,19 @@ tests = checkSequential $ Group "AD" -- build (SS (SS SZ)) (pair (pair nil 2) 3) $ #idx :-> -- oper OToFl64 $ snd_ (fst_ #idx) + snd_ #idx) - -- ,("neural", adTestGen Example.neural $ do - -- let tR = STScal STF64 - -- let genLayer nin nout = - -- liftV2 (,) <$> genArray tR (ShNil `ShCons` nout `ShCons` nin) - -- <*> genArray tR (ShNil `ShCons` nout) - -- nin <- Gen.integral (Range.linear 1 10) - -- n1 <- Gen.integral (Range.linear 1 10) - -- n2 <- Gen.integral (Range.linear 1 10) - -- input <- genArray tR (ShNil `ShCons` nin) - -- lay1 <- genLayer nin n1 - -- lay2 <- genLayer n1 n2 - -- lay3 <- genArray tR (ShNil `ShCons` n2) - -- return (input `SCons` lay3 `SCons` lay2 `SCons` lay1 `SCons` SNil)) + ,("neural", adTestGen Example.neural $ do + let tR = STScal STF64 + let genLayer nin nout = + liftV2 (,) <$> genArray tR (ShNil `ShCons` nout `ShCons` nin) + <*> genArray tR (ShNil `ShCons` nout) + nin <- Gen.integral (Range.linear 1 10) + n1 <- Gen.integral (Range.linear 1 10) + n2 <- Gen.integral (Range.linear 1 10) + input <- genArray tR (ShNil `ShCons` nin) + lay1 <- genLayer nin n1 + lay2 <- genLayer n1 n2 + lay3 <- genArray tR (ShNil `ShCons` n2) + return (input `SCons` lay3 `SCons` lay2 `SCons` lay1 `SCons` SNil)) ] main :: IO () |