diff options
| -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 () |