huwcampbell.com
💣 Machine learning which might blow up in your face 💣
Simplify recurrent example
Produced better results when I tested (the Shakespeare example already contains a more complex recurrent network, let's KISS on this one.)
authored by
koenigmaximilian
and committed by
Maximilian König
04b026f5
93435b2b
+3
-3
examples/main/recurrent.hs
+3
-3
examples/main/recurrent.hs
···
31
31
-- creation function know how to treat the layers.
32
32
type R = Recurrent
33
33
34
-
type RecNet = RecurrentNetwork '[ R (LSTM 1 4), R (LSTM 4 1)]
35
-
'[ 'D1 1, 'D1 4, 'D1 1 ]
34
+
type RecNet = RecurrentNetwork '[ R (LSTM 1 1)]
35
+
'[ 'D1 1, 'D1 1]
36
36
37
-
type RecInput = RecurrentInputs '[ R (LSTM 1 4), R (LSTM 4 1)]
37
+
type RecInput = RecurrentInputs '[ R (LSTM 1 1)]
38
38
39
39
randomNet :: MonadRandom m => m RecNet
40
40
randomNet = randomRecurrent